创建自动将整个图像数据集转换为注释文件(.xml文件)的方法

import os
import xml.etree.ElementTree as ET

def create_xml_annotation(image_dir, output_dir):
    for filename in os.listdir(image_dir):
        if filename.endswith(".jpg") or filename.endswith(".png"):
            image_path = os.path.join(image_dir, filename)
            annotation_path = os.path.join(output_dir, filename.replace(".jpg", ".xml").replace(".png", ".xml"))
            
            # 创建XML根元素
            root = ET.Element("annotation")
            
            # 创建子元素并添加标签信息
            filename_element = ET.SubElement(root, "filename")
            filename_element.text = filename
            
            # 添加其他注释信息，如图像尺寸、对象边界框等
            
            # 将XML树写入文件
            tree = ET.ElementTree(root)
            tree.write(annotation_path)