当前位置：马春杰杰 > Linux > 正文

如何批量删除VOC数据集中某个类别

发表日期：2022-05-01 马春杰杰分类：Linux 评论(0)

按照下面的格式，一个一个的删除：

#  批量移除xml标注中的某一个类别标签
import xml.etree.cElementTree as ET
import os
 
 
path_root = ['VOC2007/Annotations']
 
CLASSES = ["crowd"]   # 填入想要删去的类别，每次删去一个类
 
for anno_path in path_root:
    xml_list = os.listdir(anno_path)
    for axml in xml_list:
        path_xml = os.path.join(anno_path, axml)
        tree = ET.parse(path_xml)
        root = tree.getroot()
        print("root",root)
 
        for child in root.findall('object'):
            name = child.find('name').text
            print("name",name)
            if name in CLASSES:     # 这里可以反向写，不在Class的删掉
                root.remove(child)
        # 重写
        
        tree.write(os.path.join('VOC2007/Annotations_new', axml))  # 记得新建annotations_new文件夹

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

# 批量移除xml标注中的某一个类别标签

import xml.etree.cElementTree as ET

import os

path_root = ['VOC2007/Annotations']

CLASSES = ["crowd"] # 填入想要删去的类别，每次删去一个类

for anno_path in path_root:

xml_list = os.listdir(anno_path)

for axml in xml_list:

path_xml = os.path.join(anno_path, axml)

tree = ET.parse(path_xml)

root = tree.getroot()

print("root",root)

for child in root.findall('object'):

name = child.find('name').text

print("name",name)

if name in CLASSES: # 这里可以反向写，不在Class的删掉

root.remove(child)

# 重写

tree.write(os.path.join('VOC2007/Annotations_new', axml)) # 记得新建annotations_new文件夹

本文最后更新于2022年5月1日，已超过 1 年没有更新，如果文章内容或图片资源失效，请留言反馈，我们会及时处理，谢谢！

赞一个 (0)收藏 (0)

如果你对这篇文章有什么疑问或建议，欢迎下面留言提出，我看到会立刻回复！

打赏

未经允许不得转载：马春杰杰 » 如何批量删除VOC数据集中某个类别

标签：标签

马春杰杰

好好学习，天天向上。加油！！！TG群

相关推荐

留个评论吧~ 抢沙发评论前登陆可免验证码！

马春杰杰博客始建于2017年6月14日，博客主要分享网站建设中遇到的问题及解决方案、自己在读书过程中的心得体会及分享一些自觉有意义的音视频内容，记录些生活中的琐事，希望博客能督促怠惰的自己不断学习，不断进步。

不公开

※ 微博 ※

不公开

※ 支付宝好友 ※

不公开

※ 个人QQ号 ※

精彩直达

友情链接视频解析简历生成表白网页联系我们给我留言

TG群
TG群
微信群
微信群

订单信息（价格单位：马杜币）

*商品名称：
*商品单价：
*商品数量：

收货信息商店（虚拟商品除邮箱外可不填）

收货姓名：
收货地址：
收货邮编：
用户邮箱：
电话号码：
手机号码：
留言备注：

总金额：1.00 马杜币