YOLO数据xml文件中的一个类别名称转换为另一个类别名称

import os
import xml.etree.ElementTree as ET

# 设置包含XML文件的文件夹路径
folder_path = 'D:\\Desktop\\新建文件夹\\xml'

# 定义需要被替换的文本和替换后的文本
search_text = 'Original class'         # 替换前的类别
replace_text = 'Replacement class'           # 替换后的类别

# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
    # 检查文件扩展名是否为.xml
    if filename.endswith('.xml'):
        # 构建完整的文件路径
        file_path = os.path.join(folder_path, filename)

        # 解析XML文件
        tree = ET.parse(file_path)
        root = tree.getroot()

        # 遍历所有的<object>元素
        for obj in root.findall('object'):
            # 检查是否存在<name>元素
            name_elem = obj.find('name')
            if name_elem is not None:
                # 如果<name>元素的文本内容包含search_text，则替换文本内容
                if search_text in name_elem.text:
                    name_elem.text = name_elem.text.replace(search_text, replace_text)

        # 写入修改后的XML文件
        tree.write(file_path)

print("All XML files have been processed.")