2017-10-01 12 views
1

object detection notebookmodelsは、テスト画像上のオブジェクトを検出するためにどのように使用されるのかを示しています。ただし、ノートブック内のモデルは、COCOセットのすべてのカテゴリの検出されたオブジェクトのボックスを返します。どのようにコードを使用して、1つのカテゴリのみのオブジェクトのボックスを返すことができますか?私。モデルが確実であると確信するオブジェクトのボックスを取得するにはどうすればよいですか?人?テンソルフローオブジェクト検出で特定のカテゴリのオブジェクトのみを検出する方法

+0

オブジェクト検出コードフォルダには、視覚化のためのファイルがあります。そのファイルを編集し、関連するオブジェクト検出値のみを保持する必要があります。残りの部分を削除します(0にします)。 – Beta

+0

ありがとう、私はちょうどそれを自分自身を見つけた。すぐに改訂ノートをアップロードして、その解決方法を示します。 – user1934212

答えて

1

自分で解決策を実装しました。

https://github.com/fera0013/TrafficLightDetection/blob/master/Traffic-Light-Detection.ipynbdef filter_boxes機能をチェックして、ボックスをカテゴリ別にフィルタリングする方法を確認してください。

+0

人のカテゴリだけをフィルタリングするノートブックを実行しようとしていますが、**テスト画像のトラフィックライトを検出しています**セクションに固執しています。私はilustrate_detection(image、boxes、classes、model.split( '\\')[1]) '行に' IndexError:list index of range'を出しています。私は 'id = 1'の設定もしています。手伝って頂けますか?ありがとう! – Minoru

関連する問題