使用ultralytics8.1.3导出yolov8-obb的onnx时报错

最新推荐文章于 2025-05-31 21:56:24 发布

dyh_cy

最新推荐文章于 2025-05-31 21:56:24 发布

阅读量560

点赞数 6

文章标签： YOLO 深度学习人工智能

本文链接：https://blog.youkuaiyun.com/qq_42042528/article/details/144179961

版权

Traceback (most recent call last):
File “C:\Users\bx\Desktop\code\v8detect_1023\v8detect_1023\ultralytics\111_v8_detect_my_export官方.py”, line 5, in
model = YOLO(r"C:\Users\bx\Desktop\code\v8detect_1023\v8detect_1023\ultralytics\OBB0820v8sobb_359_1200_1024_20483.pt") # load a custom trained model
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\engine\model.py”, line 96, in init
self._load(model, task=task)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\engine\model.py”, line 162, in _load
self.model, self.ckpt = attempt_load_one_weight(weights)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\nn\tasks.py”, line 700, in attempt_load_one_weight
ckpt, weight = torch_safe_load(weight) # load ckpt
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\nn\tasks.py”, line 634, in torch_safe_load
return torch.load(file, map_location=“cpu”), file # load
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\serialization.py”, line 1025, in load
return _load(opened_zipfile,
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\serialization.py”, line 1446, in _load
result = unpickler.load()
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\serialization.py”, line 1439, in find_class
return super().find_class(mod_name, name)
AttributeError: Can’t get attribute ‘DFLoss’ on <module ‘ultralytics.utils.loss’ from ‘C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\utils\loss.py’>
使用ultralytics8.1.3导出yolov8-obb的onnx时报错，解决办法：更新ultralytics
pip install --upgrade ultralytics