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

本文介绍如何使用GitHub上的脚本将VOC2007数据集转换为COCO格式,特别指出该脚本不支持语义分割,仅适用于目标检测坐标转换,并详细记录了修改脚本的三个关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

首先进行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”否则会报类别错误

特此记录

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值