YOLO v5su模型加载报错：'Detect'对象无'grid'属性

我通过Ultralytics HUB训练了YOLO v5su模型，下载best.pt文件后本地用PyTorch运行时出现报错：'Detect' object has no attribute 'grid'。报错提示缓存可能过期，建议尝试force_reload=True，但我已尝试该方案仍无效。

完整代码 #

from PIL import Image
import torch
import cv2
model_path = "/run/media/emre/Fedora_extra/best.pt"

model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, force_reload=True)
img_path = "/run/media/emre/Fedora_extra/2d geometry figures.v3i.yolov5pytorch/valid/images/C-Shapes-project-1_jpg.rf.393826e8598965acfee9b405020e8902.jpg"

model(img_path)

完整报错信息 #

Traceback (most recent call last):
File "/home/emre/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 78, in _create
return model.to(device)
^^^^^^^^^^^^^^^^
File "/run/media/emre/Fedora_extra/anaconda/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1160, in to
return self._apply(convert)
^^^^^^^^^^^^^^^^^^^^
File "/home/emre/.cache/torch/hub/ultralytics_yolov5_master/models/common.py", line 689, in _apply
m.grid = list(map(fn, m.grid))
^^^^^^
File "/run/media/emre/Fedora_extra/anaconda/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1695, in __getattr__
raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
AttributeError: 'Detect' object has no attribute 'grid'

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "/run/media/emre/Fedora_extra/model_v5m.py", line 11, in 
model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, force_reload=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/run/media/emre/Fedora_extra/anaconda/lib/python3.11/site-packages/torch/hub.py", line 566, in load
model = _load_local(repo_or_dir, model, *args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/run/media/emre/Fedora_extra/anaconda/lib/python3.11/site-packages/torch/hub.py", line 595, in _load_local
model = entry(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^
File "/home/emre/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 88, in custom
return _create(path, autoshape=autoshape, verbose=_verbose, device=device)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/emre/.cache/torch/hub/ultralytics_yolov5_master/hubconf.py", line 83, in _create
raise Exception(s) from e
Exception: 'Detect' object has no attribute 'grid'. Cache may be out of date, try `force_reload=True` or see official documentation for help.

我是按照Ultralytics PyTorch指南编写的代码，已经尝试过force_reload=True但问题未解决，请求技术支持。

内容的提问来源于stack exchange，提问作者Mehmet Emre Demir