yolox-bytetrack-sample

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

Requirement

• OpenCV 3.4.2 or later
• onnxruntime 1.5.2 or later
• cython_bbox 0.1.3 or later

pip install -e git+https://github.com/samson-wang/cython_bbox.git#egg=cython-bbox

Demo

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

python sample.py

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

Reference

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

Author

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

License

yolox-bytetrack-sample is under MIT License.

License(Movie)

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