vehicle-reid-0001-onnx-sample

vehicle-reid-0001 を用いて、Vehicle ReIDを行うサンプルです。
モデルは同梱していますが、ダウンロードと最適化も試したい方はvehicle-reid-0001-onnx-optimize.ipynbをGoogle Colaboratoryなどで動かしてみてください。

Demo.mp4

Requirement

• OpenCV 4.8.1.78 or later
• onnxruntime 1.16.3 or later

Demo

python sample.py

• --movie_01
  視点1の動画
  デフォルト：assets/01.mp4
• --movie_02
  視点2の動画
  デフォルト：assets/02.mp4
• --yolox_model
  YOLOX（物体検出）のモデルパス
  デフォルト：yolox/model/yolox_tiny.onnx
• --yolox_score_th
  YOLOX（物体検出）のスコア閾値
  デフォルト：0.3
• --reid_model
  Vehicle ReIDのモデルパス
  デフォルト：vehicle_reid_0001/model/osnet_ain_x1_0_vehicle_reid_optimized.onnx
• --yolox_score_th
  Vehicle ReIDのスコア閾値
  デフォルト：0.5

Reference

• vehicle-reid-0001

License

vehicle-reid-0001-onnx-sampler is under Apache-2.0 license.

Author

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