本篇内容主要讲解“怎么合并PASCAL VOC的目标类别”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“怎么合并PASCAL VOC的目标类别”吧!
我们遇到数据集中需要将car、bus、truck合并成car,或将person、rider、pedestrain合并为person。使用修改标签的方法就可以实现。
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 == 'preName'): #‘preName’为修改前的名称 sku.text = 'TESTNAME' #‘TESTNAME’为修改后的名称 tree.write(file,encoding='utf-8') #写进原始的xml文件并避免原始xml中文字符乱码 else: pass else: pass if __name__ == '__main__': inputpath = 'E:/Research/Dataset/Test/Annotations' #此处替换为自己的路径
到此,相信大家对“怎么合并PASCAL VOC的目标类别”有了更深的了解,不妨来实际操作一番吧!这里是亿速云网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。