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