目标检测VOC数据集标注XML文件转EasyDL数据集标注Json格式

最新推荐文章于 2023-12-07 09:37:12 发布

SeanQikYibo

最新推荐文章于 2023-12-07 09:37:12 发布

阅读量1.8k

点赞数 1

分类专栏： Tools 文章标签：深度学习 python

本文链接：https://blog.csdn.net/qq_41672428/article/details/107451834

版权

本文介绍了如何将已有的VOC格式目标检测数据集转换为符合百度EasyDL平台要求的JSON格式，以便快速搭建目标检测模型。通过提供转换代码，将XML标注文件转化为EasyDL所需的Json格式，然后将图片和Json文件打包成Zip文件上传至平台进行训练。

摘要由CSDN通过智能技术生成

前段时间，需要使用百度大脑的EasyDL AI平台快速搭建一个目标检测模型。

平台地址：EasyDL平台

首先需要创建数据集，该平台可以自己上传数据在线标注，需要耗费较长时间。由于本人已有标注好的VOC格式的目标检测数据集，所以就想能不能直接利用起来在平台上训练。经发现可以上传有标注信息的本地数据集。

按照其示例标注json格式对VOC格式数据集中的Xml文件进行转换，代码如下：

import os
import xml.etree.cElementTree as ET
import json

#解析Xml中标注框的label和bbox
def get_bbox(xmlname):
    sig_xml_box = []
    label_name=[]
    tree = ET.parse(xmlname)
    root = tree.getroot()
    for i in root:  # 遍历一级节点
        if i.tag == 'object':
            for j in i:
                if j.tag == 'name':
                    cls_name = j.text
                    label_name.append(cls_name)
                if j.tag == 'bndbox':
                    bbox = []
                    xmin = 0
                    ymin =