EasyDL分类训练数据集Json生成

    平台地址：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'),