【数据集】在批量xml标签中选出指定标签和对应图片

最新推荐文章于 2023-11-21 15:00:35 发布

远方_boy

最新推荐文章于 2023-11-21 15:00:35 发布

阅读量3.7k

点赞数 9

分类专栏： python 数据集深度学习

本文链接：https://blog.youkuaiyun.com/weixin_41683218/article/details/81363762

版权

1.博主从朋友那里拿了一个VOC数据集，里面有20个类，博主只要6个

classes={"one","two","three","four","five","fist"}

2.选出来后把对应图片也找出来

# coding=utf-8
"""选出指定标签的xml文件"""
#"""选出对应标签的图片"""
import os
import os.path
import xml.dom.minidom
import shutil
import sys
path = "C:\\Users\\AS\\Desktop\\new\\Annotations"
newpath="C:\\Users\\AS\\Desktop\\new\\label"

label_path = "C:\\Users\\AS\\Desktop\\new\\label"
image_path = "C:\\Users\\AS\\Desktop\\new\\JPEGImages"
image_new_path ="C:\\Users\\AS\\Desktop\\new\\image"
files = os.listdir(path)  # 得到文件夹下所有文件名称
# s = []
classes={"one","two","three","four","five","fist"}
new =[]

def selete_xml_file():
    j = 1
    for xmlFile in files:
        # 遍历文件夹
        j