RT-DETR 项目【训练】【验证】【推理】脚本

文章目录

    • 训练 --train.py
    • 推理 --detect.py
    • 验证 --val.py
    • 不训练,只查看模型结构/参数量 --test.py


有同学问 RT-DETR 怎么训练,其实和 YOLOv8 几乎一样,但是有很多同学没接触过 v8

我这里直接给大家写好几个脚本,大家直接在我的脚本上调节参数就可以训练了,

脚本包含【训练】【验证】【推理】,

三个 python 文件对应三个功能 —【训练】【验证】【推理】

想魔改 RT-DETR 的同学一定先看这篇文章,可以让你少走很多弯路。

RT-DETR 项目【训练】【验证】【推理】脚本_第1张图片


训练 --train.py

RT-DETR 项目【训练】【验证】【推理】脚本_第2张图片

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

你可能感兴趣的:(RT-DETR改进实战,python,开发语言,YOLO,计算机视觉,DETR,RTDETR,人工智能)