批量修改目标检测VOC格式XML标签文件的标签名称

在目标检测任务中，可能会遇到需要修改标签名称的情况，例如修改标签类别或将多个不同的标签合并为一个。本文介绍一个Python脚本，可以批量修改VOC数据集中XML标签文件的标签名称。

功能说明

该脚本使用 xml.etree.ElementTree 库来解析和修改XML文件，通过遍历指定文件夹中的所有XML文件，将指定的标签名称替换为新的标签名称，并将修改后的文件保存到输出文件夹中。

代码实现

以下是实现该功能的代码：

import os
import xml.etree.ElementTree as ET
from tqdm import tqdm

def change_label_name(input_path, label_dict, output_path):
    """
    批量修改VOC数据集中XML标签文件的标签名称

    参数：
    - input_path: 原始XML文件夹路径
    - label_dict: 标签修改字典，键为原始标签，值为新标签
    - output_path: 修改后XML文件夹路径
    """
    xml_files = [f for f in os.listdir(input_path) if f.endswith('xml')]
    for file in tqdm(xml_files, desc="Processing XML files"):
        file_path = os.path.join(input_path, file)
        output_file_path = os.path.join(output_path, file)
        tree = ET.parse(file_path)
        root = tree.getroot()
        for obj in root.findall('object'):
            name = obj.find('name')
            if name is not None and name.text in label_dict:
                name.text = label_dict[name.text]
        tree.write(output_file_path, encoding='utf-8')

if __name__ == '__main__':
    input_path = '/path/to/original/xmls'  # 原始XML文件夹路径
    output_path = '/path/to/modified/xmls'  # 修改后XML文件夹路径

    # 标签字典，键为修改前的类别，值为修改后的类别
    label_dict = {
                  'old_class1': 'new_class1',
                  'old_class2': 'new_class2',
                  'old_class3': 'new_class3',
                  'old_class4': 'new_class4',
                  }  

    change_label_name(input_path, label_dict, output_path)

代码说明

1. input_path：原始XML文件夹路径，需要修改标签名称的XML文件所在的文件夹。
2. output_path：修改后XML文件夹路径，保存修改后XML文件的文件夹。
3. label_dict：标签字典，键为修改前的标签名称，值为修改后的标签名称。

结语

本文介绍了如何使用Python脚本批量修改目标检测VOC格式XML标签文件中的标签名称。希望对从事目标检测任务的你有所帮助。如果你有任何问题或建议，欢迎在评论区留言。