vehicle-reid-0001-onnx-sample

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

Demo.mp4

Requirement

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

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