【已解决】Recommend fixes are to train a new model using the latest ‘ultralytics‘ package

使用Ultralytics平台训练Yolov9e时报错:

TypeError: ERROR ❌️ yolov9-e.pt appears to be an Ultralytics YOLOv5 model originally trained with https://github.com/ultralytics/yolov5. This model is NOT forwards compatible with YOLOv8 at https://github.com/ultralytics/ultralytics. Recommend fixes are to train a new model using the latest ‘ultralytics’ package or to run a command with an official YOLOv8 model, i.e. ‘yolo predict model=yolov8n.pt’

训练环境:

Python 3.9
GPU RTX4090 1

训练代码:

from ultralytics import YOLO
model = YOLO("yolov9e.pt")
result = model.train(
  name="yolov9e-640",
  data = "xxx.yaml",
  epochs=500, 
  imgsz=640, 
  batch=16, 
  patience=500,
  device=0
)

解决方案:
从https://github.com/WongKinYiu/yolov9下载的模型yolov9-e.pt与本推理框架ultralytics不匹配,需要使用经过ultralytics训练的模型,模型权重下载地址:https://docs.ultralytics.com/models/yolov9/#performance-on-ms-coco-dataset

继续报错:

Traceback (most recent call last):
File “train.py”, line 2, in
model = YOLO(“yolov9e.pt”)
File “/usr/local/lib/python3.9/site-packages/ultralytics/engine/model.py”, line 94, in init
self._load(model, task)
File “/usr/local/lib/python3.9/site-packages/ultralytics/engine/model.py”, line 140, in _load
self.model, self.ckpt = attempt_load_one_weight(weights)
File “/usr/local/lib/python3.9/site-packages/ultralytics/nn/tasks.py”, line 624, in attempt_load_one_weight
ckpt, weight = torch_safe_load(weight) # load ckpt
File “/usr/local/lib/python3.9/site-packages/ultralytics/nn/tasks.py”, line 563, in torch_safe_load
return torch.load(file, map_location=‘cpu’), file # load
File “/usr/local/lib/python3.9/site-packages/torch/serialization.py”, line 809, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File “/usr/local/lib/python3.9/site-packages/torch/serialization.py”, line 1172, in _load
result = unpickler.load()
File “/usr/local/lib/python3.9/site-packages/torch/serialization.py”, line 1165, in find_class
return super().find_class(mod_name, name)
AttributeError: Can’t get attribute ‘RepNCSPELAN4’ on <module ‘ultralytics.nn.modules.block’ from ‘/usr/local/lib/python3.9/site-packages/ultralytics/nn/modules/block.py’>

解决方案:
更新ultralytics版本

pip install --upgrade ultralytics
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值