将LabelImg得到的XML文件转为CSV文件

首先,有关LabelImg这个对图像打标的软件的安装和使用方法可以参考:
https://github.com/tzutalin/labelImg#labelimg
https://blog.youkuaiyun.com/enjoy_endless/article/details/80803149
https://cloud.tencent.com/developer/news/325876

本文主要讲解,如何将LabelImg软件标注图片后得到的XML文件,转化为对应的CSV文件数据的代码详解。

一:文件环境说明
首先展示我的操作和代码环境。
(1)在my_test文件夹下,有images文件夹和转换代码xml_to_csv.py
以下my_test文件夹下,就是我们的工作根目录
在这里插入图片描述
(2)images文件夹下有test文件夹和train文件夹,分别存在我们的测试图片集和训练图片集在这里插入图片描述
(3)经过LabelImg标注后,得到对应的XML文件,如下所示

LabelImg是一款流行的图像标注工具,它可以生成XML格式的标注文件,这些文件包含了图像中的物体边界框位置和类别信息。将LabelImg生成的XML文件化为CSV文件,可以通过编写脚本来实现,下面是一个简单的Python脚本示例,用于将XML格式的标注文件换为CSV文件: ```python import os import glob import pandas as pd from xml.etree import ElementTree as ET def convert(size, box): dw = 1. / size[0] dh = 1. / size[1] x = (box[0] + box[1]) / 2.0 y = (box[2] + box[3]) / 2.0 w = box[1] - box[0] h = box[3] - box[2] x = x * dw w = w * dw y = y * dh h = h * dh return (x, y, w, h) def convert_annotation(xml_file, csv_file): tree = ET.parse(xml_file) root = tree.getroot() size = root.find('size') w = int(size.find('width').text) h = int(size.find('height').text) with open(csv_file, 'w', newline='') as file: csv_file.write('filename,x1,y1,x2,y2,class\n') for member in root.findall('object'): bndbox = member.find('bndbox') xmin = int(bndbox.find('xmin').text) ymin = int(bndbox.find('ymin').text) xmax = int(bndbox.find('xmax').text) ymax = int(bndbox.find('ymax').text) class_name = member.find('name').text xy = convert((w, h), (xmin, ymin, xmax, ymax)) csv_file.write(f'{os.path.basename(xml_file)},{xy[0]},{xy[1]},{xy[2]},{xy[3]},{class_name}\n') # 假设xml_dir是包含所有XML标注文件文件夹路径,csv_dir是CSV文件保存的文件夹路径 xml_dir = 'path/to/xmls' csv_dir = 'path/to/csvs' # 获取所有XML文件的路径 xml_files = glob.glob(os.path.join(xml_dir, "*.xml")) # 遍历XML文件换为CSV格式 for xml_file in xml_files: csv_file = os.path.join(csv_dir, os.path.basename(xml_file)[:-3] + 'csv') convert_annotation(xml_file, csv_file) ``` 在使用上述脚本之前,请确保已经安装了`pandas`和`lxml`(用于解析XML文件)库。如果还没有安装,可以通过以下命令安装: ```bash pip install pandas lxml ``` 脚本中`convert_annotation`函数负责解析单个XML文件,并将其换为一行CSV格式的数据。然后,通过遍历一个包含所有XML文件文件夹,我们可以处理所有的图像标注文件
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值