python批量修改voc数据集内容

最新推荐文章于 2023-08-15 21:12:14 发布

darkredrock

最新推荐文章于 2023-08-15 21:12:14 发布

阅读量344

点赞数 1

分类专栏： python小程序文章标签： python

本文链接：https://blog.csdn.net/darkredrock/article/details/126796097

版权

python小程序专栏收录该内容

1 篇文章 0 订阅

订阅专栏

在多个voc数据合并在一起的时候，有时候会存在数据集标注的class分类不同，所以需要更换类别名称，此代码功能用于搜索需要更改的name。


import os
import xml.etree.ElementTree as ET
 
#程序功能：批量修改VOC数据集中xml标签文件的标签名称
def changelabelname(inputpath):
    listdir = os.listdir(inputpath)
    for file in listdir:
        if file.endswith('xml'):
            file = os.path.join(inputpath,file)
            tree = ET.parse(file)
            root = tree.getroot()
            for object1 in root.findall('object'):
                for sku in object1.findall('name'):           #查找需要修改的名称
                    if (sku.text == 'person'):               #‘preName’为修改前的名称
                        sku.text = 'headshoulder'                 #‘TESTNAME’为修改后的名称
                        print('更新成功')
                        tree.write(file,encoding='utf-8')     #写进原始的xml文件并避免原始xml中文字符乱码
                
 
if __name__ == '__main__':
    inputpath = 'C:\\Users\\Administrator\\Desktop\\Annotations_person\\'  #此处替换为自己的路径
    changelabelname(inputpath)

转载：提取voc中需要的类别以及修改类名称_tian-happy的博客-CSDN博客