Awesome
Warning<br> このリポジトリはNanoDet-Plusには対応していません。
NanoDet-ONNX-Sample
NanoDetのPythonでのONNX推論サンプルです。<br> ONNXに変換したモデルも同梱しています。変換自体を試したい方はConvert2ONNX.ipynbを使用ください。<br>
Requirement
- OpenCV 3.4.2 or later
- onnxruntime 1.5.2 or later
- Pytorch 1.7 or later ※ONNX変換を実施する場合のみ
- pytorch-lightning 1.4.7 or later ※ONNX変換を実施する場合のみ
Demo
デモの実行方法は以下です。
python sample_onnx.py
- --device<br> カメラデバイス番号の指定<br> デフォルト：0
- --movie<br> 動画ファイルの指定 ※指定時はカメラデバイスより優先<br> デフォルト：指定なし
- --image<br> 画像ファイルの指定 ※指定時はカメラデバイスや動画より優先<br> デフォルト：指定なし
- --width<br> カメラキャプチャ時の横幅<br> デフォルト：960
- --height<br> カメラキャプチャ時の縦幅<br> デフォルト：540
- --model<br> ロードするモデルの格納パス<br> デフォルト：model/nanodet_m_320.onnx
- --input_shape<br> モデルの入力サイズ<br> デフォルト：320
- --score_th<br> クラス判別の閾値<br> デフォルト：0.35
- --nms_th<br> NMSの閾値<br> デフォルト：0.6
Reference
Author
高橋かずひと(https://twitter.com/KzhtTkhs)
License
NanoDet-ONNX-Sample is under Apache-2.0 License.
License(Movie)
サンプル動画はNHKクリエイティブ・ライブラリーのイギリス ウースターのエルガー像を使用しています。