目录
1.数据集准备
1.1建立自己的数据集
为了快速验证训练可行,我只选10张图片参与训练过程(5张train,4张value,1张test)URPC标注存储为XML
,为了方便操作,我选择PascalVOC数据格式来训练(detectron2能处理COCO、PascalVOC等等格式,具体还支持哪些数据格式可以参考detectron2的源码)
1.1.1接下来建议在本地将数据集准备好
按下图新建几个文件夹
-VOC2007
-Annotations
-[放入打好标签的XML文件]
-ImageSets
-Main
-test.txt
-train.txt
-val.txt
-JPEGImages
-[放入数据集图片]
1.2将数据集部署到服务器
按照格式建立好打成ZIP格式上传至服务器放在DiffusionDet-main/datasets
路径下然后解压
2.准备预训练模型
https://github.com/ShoufaChen/DiffusionDet
我这里选择的是COCO-Res50
并把它放到DiffusionDet-Main/model(自己建的)路径下