关于自己 run sh gen_det.sh ${MCMT_CONFIG_FILE}

[Fish39171]

出现以下Error:
image 1/2001 /home/fish/AIC21-MTMC/datasets/detection/images/test/S06/c043/img1/img000000.jpg: Traceback (most recent call last):
File "detect2img.py", line 235, in
detect()
File "detect2img.py", line 95, in detect
pred = model(img, augment=opt.augment)[0]
File "/home/fish/anaconda3/envs/aic21/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/fish/AIC21-MTMC/detector/yolov5/models/yolo.py", line 119, in forward
return self.forward_once(x, profile) # single-scale inference, train
File "/home/fish/AIC21-MTMC/detector/yolov5/models/yolo.py", line 135, in forward_once
x = m(x) # run
File "/home/fish/anaconda3/envs/aic21/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/fish/AIC21-MTMC/detector/yolov5/models/yolo.py", line 54, in forward
y[..., 2:4] = (y[..., 2:4] * 2) ** 2 * self.anchor_grid[i] # wh
RuntimeError: The size of tensor a (160) must match the size of tensor b (48) at non-singleton dimension 3

想请问一下我应该如何处理？

[LCFractal]

yolov5没有进行修改，现在的yolov5版本可能已经和之前的不一致了，请尝试我们在Google网盘里提供的yolov5x.pt预训练模型。你也可以尝试使用最新的yolov5代码替换检测器部分

[Fish39171]

感谢解答