解决数据集中xml文件类别标签的首字母大小写不一致问题

方方爱学习

于 2024-11-05 19:22:07 发布

阅读量276

点赞数 3

CC 4.0 BY-SA版权

分类专栏：笔记 CV学习笔记文章标签： xml 重构 YOLO python 人工智能

本文链接：https://blog.csdn.net/weixin_45920120/article/details/143522534

import os
import xml.etree.ElementTree as ET

# 指定要处理的 XML 文件夹路径
xml_folder_path = r'D:\CVproject\ultralytics-main\datatrans\Annotations'


def capitalize_first_letter_in_xml(xml_file):
    # 解析 XML 文件
    tree = ET.parse(xml_file)
    root = tree.getroot()

    # 遍历所有 <name> 标签
    for name_elem in root.findall('.//name'):
        original_name = name_elem.text
        if original_name and original_name[0].islower():
            # 将首字母大写
            modified_name = original_name.capitalize()
            name_elem.text = modified_name
            print(f"Updated '{original_name}' to '{modified_name}' in file '{xml_file}'")

    # 将修改后的内容写回原文件
    tree.write(xml_file, encoding='utf-8', xml_declaration=True)


if __name__ == "__main__":
    # 获取文件夹下所有 XML 文件
    for filename in os.listdir(xml_folder_path):
        if filename.endswith('.xml'):
            file_path = os.path.join(xml_folder_path, filename)
            capitalize_first_letter_in_xml(file_path