RT-DETR使用教程（超越YOLO的实时目标检测器）

阅读本书更多章节>>>> 2.1 模型下载

      作者版本下载链接：https://github.com/lyuwenyu/RT-DETR

      本文教程所用版本链接：https://gitcode.com/ultralytics/ultralytics

 2.1 创建训练脚本

        新建一个训练脚本，命名为train,并填入图中代码，运行即可。

from ultralytics.models import RTDETRif __name__ == '__main__':    model = RTDETR(model='ultralytics/cfg/models/rt-detr/rtdetr-l.yaml')    model.load('rtdetr-l.pt')    model.train(pretrained=True, data='data.yaml', epochs=1, batch=1, device=0, imgsz=320, workers=2,cache=False,)

---

马上推出RT-DETR改进项目，欢迎关注！

---

阅读本书更多章节>>>>

本文链接：https://www.kjpai.cn/gushi/2024-05-04/164685.html，文章来源：网络cs，作者：璐璐，版权归作者所有，如需转载请注明来源和作者，否则将追究法律责任！