基于u版的yolov3/yolov5上搭建自己的模型，修改主干網絡或者head，訓練可以正常訓練，訓練完成后用命令python val.py --weights runs/exp/train/best.pt進行測試，報錯：
RuntimeError: Sizes of tensors must match except in dimension 2. Got 27 and 28 (The offending index is 0)
如下圖所示

開始以為是維度錯誤或者特征圖大小不匹配，但能正常訓練，應該不是搭建的網絡有問題，最后在github/yolov5找到了解決方法：
定位common.py，將#stride = int(model.stride.max())修改為stride = max(int(model.stride.max()), 32)，如下：

elif pt: # PyTorchfrom models.experimental import attempt_load # scoped to avoid circular importmodel = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location=device)#stride = int(model.stride.max()) # model stridestride = max(int(model.stride.max()), 32) # model stride

總結

以上是生活随笔為你收集整理的yolov3/yolov5 val.py训练正常测试错误的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。