TensorFlow Object Detection APIのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。
変換自体を試したい方はObject-Detection-API-TensorFlow2ONNX.ipynbをColaboratoryで実行してください。
TensorFlow-Object-Detection-API-ONNX-Demo.mp4
- OpenCV 3.4.2 or later
- onnxruntime 1.5.2 or later
デモの実行方法は以下です。
python demo.py
- --device
カメラデバイス番号の指定
デフォルト:0 - --movie
動画ファイルの指定 ※指定時はカメラデバイスより優先
デフォルト:指定なし - --width
カメラキャプチャ時の横幅
デフォルト:960 - --height
カメラキャプチャ時の縦幅
デフォルト:540 - --model
ロードするモデルの格納パス
デフォルト:model/efficientdet_d0.onnx - --input_shape
モデルの入力サイズ
デフォルト:512,512 - --score_th
クラス判別の閾値
デフォルト:0.5
高橋かずひと(https://twitter.com/KzhtTkhs)
Object-Detection-API-TensorFlow2ONNX is under Apache-2.0 License.
サンプル動画はNHKクリエイティブ・ライブラリーのイギリス ウースターのエルガー像を使用しています。