示例EasyDL的分类json格式如下:
{"labels": [{"name": "Drama"}, {"name": "Romance"}]}
按照其示例标注json格式对VOC格式数据集中的Xml文件进行转换,代码如下:
import os import xml.etree.cElementTree as ET import json data_dir = r'./Test' for dir in os.listdir(data_dir): class_dir = os.path.join(data_dir,dir) img_list = os.listdir(class_dir) for xml_name in img_list: json_name = str(xml_name.split('.')[0]) + '.json' json_path = os.path.join(class_dir, json_name) ann_js = {} annotations = [] anno = {} anno['name'] = dir #这里文件夹名即为类名 annotations.append(anno) ann_js['labels'] = annotations json.dump(ann_js, open(json_path, 'w'), indent=4) # indent=4 更加美观显示
然后将原图和Json文件打包为Zip,压缩包要求的格式如下:
打包上传至数据中心即可开始训练。
EasyDL目标检测数据集制作可跳至:目标检测VOC数据集标注XML文件转EasyDL数据集标注Json格式