DETR训练自己的数据集
本文记录利用官方提供的代码来训练验证自己的数据集的过程以及一些注意事项。该数据集原始的标签保存格式为yolo类型的txt文件,下面提供的代码能够实现数据集标签从txt文件到coco json格式的转化.该部分为利用训练得到的模型权重进行检测验证。在detr工程文件下的images文件夹存放所有待验证的图片,infer_results存放所有的验证结果。DETR需要的数据集格式为coco格式,其原始图片与标签分为训练集和验证集,其保存目录为。output_dir为保存训练得到的模型权重的地址。






