VOC转coco脚本,仅用于目标检测,非分割任务

首先进行VOC2007转为coco数据集的格式:
从github上下载voc转coco的脚本,该脚本不包含语义分割的segmentation,只能用于目标检测的坐标转换
下载的传送门:https://github.com/jiachen0212/voc2coco-pattern
修改其中的xml2json.py脚本
(1)voc_clses= …(此处加上__background__)即填写n加1类的类名
(2)修改cat_id=voc_clses.index(cls_name) + 1,将后面的加1去掉,否则会出现category的id与annotations中的id相隔1的问题,此问题卡了很久,该处的错误会导致后续的json转换tfrecord时的label无法匹配,出现问题,只录入坐标而不录入类名
(3)将json的name注释掉,重新写入
jStr=json.dump(ann_js,ensure_ascii =False)
with open (“train.json”,“w”)as f:
f.write(jStr)
不这样写会报错,train.json不存在,同时with语句中的“w”不可以为“wb”否则会报类别错误

特此记录

你可能感兴趣的:(object,detection)