平台地址:EasyDL平台
示例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'),