Kazuhito00 / Object-Detection-API-TensorFlow2ONNX

TensorFlow Object Detection APIのONNX推論サンプル

Object-Detection-API-TensorFlow2ONNX

TensorFlow Object Detection APIのONNX推論サンプルです。
ONNXに変換したモデルも同梱しています。
変換自体を試したい方はObject-Detection-API-TensorFlow2ONNX.ipynbをColaboratoryで実行してください。

TensorFlow-Object-Detection-API-ONNX-Demo.mp4

Requirement

• OpenCV 3.4.2 or later
• onnxruntime 1.5.2 or later

Demo

デモの実行方法は以下です。

python demo.py

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540
• --model
  ロードするモデルの格納パス
  デフォルト：model/efficientdet_d0.onnx
• --input_shape
  モデルの入力サイズ
  デフォルト：512,512
• --score_th
  クラス判別の閾値
  デフォルト：0.5

Reference

• TensorFlow Object Detection API

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

Object-Detection-API-TensorFlow2ONNX is under Apache-2.0 License.

License(Movie)

サンプル動画はNHKクリエイティブ・ライブラリーのイギリス ウースターのエルガー像を使用しています。