ByteTrack与Yolov5

Question

在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'

我怎么才能解决这个问题？

ByteTrack回购

Yovo5回购

Answer 1

问题是，您试图将Yolov5权重加载到YoloX体系结构中。您不能互换地使用YoloX/Yolov5 5资源。如果要使用YoloX，则必须将与后者相关的所有内容替换为对应的YoloV5。