需求： ? ? ? ? 數(shù)據(jù)集的數(shù)據(jù)增強中，有時需要將xml標注文件中的類別做修改為新類別，或者將幾個類別合并為一個類別。 ? 解決方法： ? ? ? ? 使用python+import xml.etree.ElementTree將xml標注文件中的類別名稱做修改。代碼如下：

import os
import glob
import xml.etree.ElementTree as et

def change_labels(source_dir):
    file_list = glob.glob(os.path.join(source_dir,"*.xml"))
    print(len(file_list))
    index = 1
    for file in file_list:
        print(file,index)
        index += 1
        tree_ = et.ElementTree()
        tree_.parse(file)

        root = et.Element("annotation")
        folder = et.SubElement(root, "folder")
        folder.text = "images"
        filename = et.SubElement(root, "filename")
        filename.text = tree_.find(".//filename").text
        path = et.SubElement(root, "path")
        path.text = "/home/Desktop/labels"
        source = et.SubElement(root, "source")
        database = et.SubElement(source, "database")
        database.text = "Unknown"
        size = et.SubElement(root, "size")
        width = et.SubElement(size, "width")
        width.text = tree_.find(".//width").text
        height = et.SubElement(size, "height")
        height.text = tree_.find(".//height").text
        depth = et.SubElement(size, "depth")
        depth.text = "3"
        segmented = et.SubElement(root, "segmented")
        segmented.text = "0"

        objects = tree_.findall(".//object")
        anchor = 0
        for object in objects:

            try:
                name_ = object.find("name").text
            except:
                print(file)
                print("-----------------------------------------")
                continue
            
            n = ""
            if name_ == "van" or name_ == "bus" or name_ == "truck":
                n = "car"
            else:
                n = name_
            object_ = et.SubElement(root, "object")
            name = et.SubElement(object_, "name")
            name.text = n
            pose = et.SubElement(object_, "pose")
            pose.text = "Unspecified"
            truncated = et.SubElement(object_, "truncated")
            truncated.text = "0"
            difficult = et.SubElement(object_, "difficult")
            difficult.text = "0"
            bndbox = et.SubElement(object_, "bndbox")
            xmin = et.SubElement(bndbox, "xmin")
            xmin.text = object.find(".//xmin").text
            ymin = et.SubElement(bndbox, "ymin")
            ymin.text = object.find(".//ymin").text
            xmax = et.SubElement(bndbox, "xmax")
            xmax.text = object.find(".//xmax").text
            ymax = et.SubElement(bndbox, "ymax")
            ymax.text = object.find(".//ymax").text

        tree = et.ElementTree(root)
        tree.write(file,encoding="utf-8")

if __name__ == '__main__':
    source_dir = "/home/Desktop/labels/Annotations"
    flip_images(source_dir)

?使用以上代碼需要修改原圖像和標注文件所在文件夾路徑（source_dir），也需要修改類別，上述代碼將"van”、“bus”、“truck”修改為“car”，這里可根據(jù)實際情況自行修改。親測可用。?

文章來源地址http://www.zghlxwxcb.cn/news/detail-683868.html

到了這里，關(guān)于目標檢測項目中，使用python+xml.etree.ElementTree修改xml格式標注文件中的類別名稱的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！