YOLOPのPythonでのONNX推論サンプル

Overview

YOLOP-ONNX-Video-Inference-Sample

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

Requirement

  • OpenCV 3.4.2 or later
  • onnxruntime 1.9.0 or later

Demo

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

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

Reference

Author

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

License

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

License(Movie)

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

You might also like...
Owner
KazuhitoTakahashi
KazuhitoTakahashi