iOSでTensorflow Object Detection APIモデルを実行するにはどうすればよいですか？

Question

満足のいく結果を得たモデルを訓練したばかりで、frozen_inference_graph.pbを持っています。 iOSでこれを実行するにはどうすればいいですか？それはSSD Mobilenet V1で訓練されています。最適にはGPUを使って実行したいのですが（私はテンソルフローAPIがiOS上でそれを行うことはできませんが）、最初にCPU上に置くのは素晴らしいことです。

Answer 1

コアMLにTensorFlowモデルをインポートするためのサポートはjust announcedでした。これは、the tfcoreml converterを使用して実行されます。これは、.pbグラフを取り込んでCore MLモデルを出力する必要があります。そこから、Core MLでこのモデルを使用し、処理のために静止画像またはビデオフレームを取り込むことができます。

正しい入力色空間とサイズを提供していることを確認してから、SSD結果を正しく抽出して処理して、オブジェクトクラスと境界ボックスを取得する必要があります。