![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
transformer
文章平均质量分 77
githubcurry
githubcurry
展开
-
使用DETR 训练VOC数据集和自己的数据集
DETR用的是的数据集。LabelimgCOCO数据集的格式类似这样,annotations文件夹里面有对应的train、val数据集的json文件。train2017则是训练集图片,其他同理。VOC数据集的存放方式是这样的,转换格式就是找出Main文件夹下用于目标检测的图片。Main文件夹下有train.txt文件,记录了训练集的图片。val.txt记录了验证集的图片只需要修改注释中的两个路径即可(创建文件夹时没有加判断语句严谨一点应该加上)。原创 2023-12-18 10:33:08 · 1069 阅读 · 0 评论 -
如何用DETR(detection transformer)训练自己的数据集,并推理图片
这里我分享一下官方写的DETR的预测代码,其实就是github上DETR官方写的一个Jupyter notebook,可能需要梯子才能访问,这里我贴出来。因为DETR比较难训练,我觉得可以用官方的预训练模型来看看效果。顺便说一下,这个模型是简易版,其实就是DETR论文最后一页贴出来的代码,mAP大概要比源码的低2-3个点,但已经很强了。只用了50行代码就把DETR模型架构展示出来了,真的是非常简洁的一个模型!原创 2023-12-13 10:29:26 · 621 阅读 · 0 评论