文章目录
- 训练 --train.py
- 推理 --detect.py
- 验证 --val.py
- 不训练,只查看模型结构/参数量 --test.py
 
有同学问 RT-DETR 怎么训练,其实和 YOLOv8 几乎一样,但是有很多同学没接触过 v8
我这里直接给大家写好几个脚本,大家直接在我的脚本上调节参数就可以训练了,
脚本包含【训练】【验证】【推理】,
三个 python 文件对应三个功能 —【训练】【验证】【推理】
想魔改 RT-DETR 的同学一定先看这篇文章,可以让你少走很多弯路。

训练 --train.py

大家可以在 ultralytics 文件夹下新建一个train-rtdetr.py,然后直