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