手把手教你使用云服务器训练 RT-DETR (Pytorch版)

大家好,我是迪菲赫尔曼,这篇文章我将手把手带大家训练 RT-DETR 目标检测模型,

这篇教程是为刚入门同学准备的,如果你不会配置环境,又没有像样的算力,那我推荐大家严格跟着我的教程,很多坑我已经给大家通过代码补上了,跟着我的教程肯定能跑通。

我将采用 VSCode 远程连接 AutoDL 的方式进行本次教学, 代码采用 ultralytics 团队开源的 YOLOv8 项目进行训练,这个项目目前支持 RT-DETR 的训练验证和推理!

首先来说数据集, RT-DETR 的数据集格式是标准的 YOLO格式的数据集,也就是一张图片对应一个标签,标签中有五个维度的信息,分别代表:物体的类别,物体框的中心点x坐标,中心点y坐标,框的宽,框的高。

虽然当我们训练时会

相关文章

评论可见,请评论后查看内容,谢谢!!!评论后请刷新页面。