import os
import xml.etree.ElementTree as ET

# 你的 XML 文件所在的文件夹
ANNOTATION_DIR = "/home/lab/LJ/wampee/ssd-pytorch-master/VOCdevkit/VOC2007/Annotations"

# 遍历 `Annotations` 目录下的所有 XML 文件
for xml_file in os.listdir(ANNOTATION_DIR):
    if not xml_file.endswith(".xml"):
        continue  # 跳过非 XML 文件

    xml_path = os.path.join(ANNOTATION_DIR, xml_file)
    tree = ET.parse(xml_path)
    root = tree.getroot()

    # 找到 <filename> 标签
    filename_tag = root.find("filename")
    if filename_tag is not None:
        filename = filename_tag.text
        base_name, ext = os.path.splitext(filename)  # 分离文件名和扩展名

        if ext.lower() == ".png":
            # 修改后缀为 .jpg
            new_filename = base_name + ".jpg"
            filename_tag.text = new_filename

            # 保存修改后的 XML
            tree.write(xml_path, encoding="utf-8")
            print(f"✅ 修改 {xml_file}: {filename} → {new_filename}")

print("🎉 所有 XML 文件修改完成！")