车辆非法掉头检测

YOLOX+ ByteTrack+ Trajectory Analysis

Intro

YOLOX原模型用于目标检测，ByteTrack用于关联目标，最后进行简单轨迹分析来预测车辆是否掉头。
A simple extension of YOLOX and ByteTrack. Use YOLOX model for vehicle(car, bus, truck) detection, ByteTrack for association. And a simple trajectory analysis is used for 'u-turn' detection.

Tracking Demo

Installation

refer to YOLOX

Model Weights

yolox_m

Video

original video
results

Demo

python tools/demo_vehicle_track.py video -f exps/example/yolox_m_vehicle_det.py -c ./pretrained/yolox_m.pth --path ./videos/test_13.mp4 --conf 0.25 --nms 0.45 --tsize 640 --save_result --device gpu

Track other objects (COCO dataset, 80 classes)

pls modify

./exps/example/yolox_m_vehicle_det.py  # self.track_classes