rt-detrv2训练自己数据集（官方版|数据为coco格式）
rt-detrv2训练自己数据集（coco格式）
·
参考这两位大佬：
【代码复现】RT-DETR2（pytorch版）环境部署及训练_rt-detrv2-CSDN博客
RTDETRv2 pytorch训练_rt-detr训练数据集csdn-CSDN博客
1、我用的这个，里面这样的，不用改
2、改刚刚那个里面写的几个路径对应的文件，首先是这个，输入数据集路径，num_classes这个，我是yolo有0123456，转成coco转到1234567，这个地方写7
3、改这个，这个optimizer里面是有一个epoch，如果要改就改下
要是改了上面的，下面的也要改一样
4、tools/train.py
改成这样，本来后面还有一个XX=true，要删掉。
5、python train.py即可
训练生成的在tools/output里面，有best,last这样的
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
5
0- 0
已为社区贡献2条内容
所有评论(0)