yolox-motpy-sample

YOLOXとmotpyを用いたMOT(Multiple Object Tracking)のPythonサンプルです。
YOLOXはYOLOX-ONNX-TFLite-Sampleで、ONNXに変換したモデルを使用しています。

08-01.yolox-motpy.mp4

Requirement

• OpenCV 3.4.2 or later
• onnxruntime 1.5.2 or later
• filterpy 1.4.5 or later

Demo

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

python sample.py

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --width
  カメラキャプチャ時の横幅
  デフォルト：960
• --height
  カメラキャプチャ時の縦幅
  デフォルト：540

YOLOXパラメータ

• --yolox_model
  ロードするモデルの格納パス
  デフォルト：model/yolox_nano.onnx
• --input_shape
  モデルの入力サイズ
  デフォルト：416,416
• --score_th
  クラス判別の閾値
  デフォルト：0.3
• --nms_th
  NMSの閾値
  デフォルト：0.45
• --nms_score_th
  NMSのスコア閾値
  デフォルト：0.1
• --with_p6
  Large P6モデルを使用するか否か
  デフォルト：指定なし

motpyパラメータ

• --max_staleness
  デフォルト：5
• --order_pos
  デフォルト：1
• --dim_pos
  デフォルト：2
• --order_size
  デフォルト：0
• --dim_size
  デフォルト：2
• --q_var_pos
  デフォルト：5000.0
• --r_var_pos
  デフォルト：0.1
• --tracker_min_iou
  デフォルト：0.25
• --multi_match_min_iou
  デフォルト：0.93
• --min_steps_alive
  デフォルト：3

※パラメータ詳細はmotpyを参照ください。

Reference

• Megvii-BaseDetection/YOLOX
• YOLOX-ONNX-TFLite-Sample
• motpy

Author

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

License

yolox-motpy-sample is under MIT License.

License(Movie)

サンプル動画はNHKクリエイティブ・ライブラリーのケニア共和国キツイ 町並み(4) ふかんショットを使用しています。