YOLOPのPythonでのONNX推論サンプルです。
ONNXモデルは、hustvl/YOLOP/weights を使用しています。

• OpenCV 3.4.2 or later
• onnxruntime 1.9.0 or later

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

python sample.py

• --video
  動画ファイルの指定
  デフォルト：video/sample.mp4
• --model
  ロードするモデルの格納パス
  デフォルト：weights/yolop-640-640.onnx
• --input_size
  モデルの入力サイズ
  デフォルト：640,640
• --score_th
  クラス判別の閾値
  デフォルト：0.3
• --nms_th
  NMSの閾値
  デフォルト：0.45

• hustvl/YOLOP

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

YOLOP-ONNX-Video-Inference-Sample is under MIT License.

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