Байт-трек с Yolov5

#python #computer-vision #yolov5

#python #компьютерное зрение #yolov5

Вопрос:

На этапе обучения ByteTrack я хочу использовать Yolov5 вместо YoloX, поэтому я пытаюсь запустить

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

с предварительно подготовленными весами yolov5x6 из модельного зоопарка yolov5, и я получил эту ошибку

ModuleNotFoundError: No module named 'models'

Как я могу это исправить?
Репозиторий байт-трека репозитория
Yovo5