网上看了很多教程,大概都是需要改断点训练,或者使用最新版本Ultralytics,但作者尝试很久,这些方法都不能成功,最后去github论坛才发现导入有问题。

我最初一直是

from ultralytics import YOLO
def train():
    model = YOLO(model=r"E:\YOLO\yolo8\ultralytics\cfg\models\rt-detr\rtdetr-l.yaml", task="detect")

后面改为

from ultralytics import RTDETR
def train():
   
    model = RTDETR(model=r"E:\YOLO\ultralytics-main\ultralytics\cfg\models\rt-detr\rtdetr-l.yaml")

成功运行!!!!

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐