将LabelImg图像标注文件转化为csv文件

LabelImg是一款流行的图像标注工具，它可以生成XML格式的标注文件，这些文件包含了图像中的物体边界框位置和类别信息。将LabelImg生成的XML文件转化为CSV文件，可以通过编写脚本来实现，下面是一个简单的Python脚本示例，用于将XML格式的标注文件转换为CSV文件：

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文件）库。如果还没有安装，可以通过以下命令安装：

pip install pandas lxml

脚本中`convert_annotation`函数负责解析单个XML文件，并将其转换为一行CSV格式的数据。然后，通过遍历一个包含所有XML文件的文件夹，我们可以处理所有的图像标注文件。