ByteTrack with Yolov5

Viewed 442

In the training step of ByteTrack, I want to use Yolov5 instead of YoloX, so I try to run

python3 tools/train.py -f exps/example/mot/yolox_x_mot17_half.py -o -c pretrained/yolov5x6.pt -b 1 -d 1

with yolov5x6 pretrained weights from yolov5 model zoo, and I got this error

ModuleNotFoundError: No module named 'models'

How can I fix this?
ByteTrack repo
Yovo5 repo

0 Answers
Related