python 将xml汇总转换为CSV格式

最新推荐文章于 2023-08-30 11:27:21 发布

读书小孩

最新推荐文章于 2023-08-30 11:27:21 发布

阅读量2.4k

点赞数 2

分类专栏：图像标注

本文链接：https://blog.csdn.net/qq_43061705/article/details/105122035

版权

本文提供Python代码示例，将labelimg和标注精灵两款标注工具的XML数据转换成CSV格式。请注意，根据实际XML结构进行适当调整。

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

注意：下面代码主要针对两种不同的xml写的，根据自身情况修改（分别是labelimg、标注精灵两款标注工具）

#解决了一个xml多个标签问题
import os
import glob
import pandas as pd
import xml.etree.ElementTree as ET
"""两种不同标准工具的xml转换为CSV（两种标注出来的xml格式不一样）"""
def xml_to_csv(path):
    xml_list = []
    a=0
    print(glob.glob(path + '/*.xml'))
    for xml_file in glob.glob(path + '/*.xml'):
        tree = ET.parse(xml_file)
        root = tree.getroot()
        a+=1
        print("地址：{}".format(xml_file),"第{}个".format(a))

        if root.findall('object'):##labelimg标注工具
            print("labelimg")
            for member in root.findall('object'):
                print(root.find('filename').text,
                         int(root.find('size')[0].text),
                         int(root.find('size')[1].text),