YOLOPv2-ONNX-Sample

YOLOPv2のPythonでのONNX推論サンプルです。
ONNX変換自体を試したい方はColaboratoryでConvert2ONNX.ipynbを使用ください。

Demo.mp4

Requirement

OpenCV 3.4.2 or later
onnxruntime 1.9.0 or later

Demo

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

python sample.py

--video
動画ファイルの指定
デフォルト：video/sample.mp4
--model
ロードするモデルの格納パス
デフォルト：weight/YOLOPv2.onnx ※ファイルが無い場合ダウンロードを行います
--input_size
モデルの入力サイズ
デフォルト：640,640
--score_th
クラス判別の閾値
デフォルト：0.3
--nms_th
NMSの閾値
デフォルト：0.45

Reference

CAIC-AD/YOLOPv2

Author

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

License

YOLOPv2-ONNX-Sample is under MIT License.

License(Movie)

サンプル動画はNHKクリエイティブ・ライブラリーの中国・重慶　高速道路を走るを使用しています。