出发点:DETR数据格式的需要。
这几天一直在做DETR目标检测,第一步就是要求将标注好的数据集转化为coco格式,因为我也是萌新,找了好久才看到一篇文章写的言简意赅,亲测有效。为了避免自己后面忘记怎么做所以特意记录一下,我相信也有很多和我一样刚接触这方面的朋友,刚上来看各种大佬的文章整的各种焦头烂额,所以在下不才,也为后人提供一个方向,我参考的文章具体链接在这里:LabelMe标记后的json文件怎么转为COCO格式_json转coco-CSDN博客1
1、下载和安装LabelMe,https://github.com/wkentaro/labelme
2、将原始图片和用LabelMe标注好的json文件放在同一个文件夹内,如下所示:
3、打开Anaconda Prompt,进入到 labelme/examples/instance_segmentation/路径下,具体做法为cd 目标文件路径,如下图(cd后面改为自己的文件路径)
cd D:\com_visionExperiments\Design\labelme-master\labelme-master\examples\instance_segmentation
需要注意的是第二步里面的图片和json文件也要放在instance_segmentation路径下。
4、找到给文件路径下的label.txt文件将里面的类别修改为自己的数据需要检测的类别,如下图,(我这里只有三种检测类别)
5、在Anaconda_Prompt里运行如下代码,注意这里需要提前安装pycocotools,具体操作为pip install pycocotools
python labelme2coco.py coco_root/ output --labels labels.txt
这里的coco_root是存放原始图片和标注好的json文件夹名,output是存放转化结果的文件夹名,其中output不用新建。运行结果如下: