python挑出训练集里图片对应的xml文件，方便统计标签框的类别与数目

最新推荐文章于 2024-04-15 16:51:05 发布

_一点点_

最新推荐文章于 2024-04-15 16:51:05 发布

阅读量1.5k

点赞数

分类专栏： Python基本操作文章标签： python

本文链接：https://blog.csdn.net/weixin_41991401/article/details/89517694

版权

本文介绍如何使用Python解析XML文件，针对训练集中的图像，统计各类标签框的数量，为机器学习模型的训练提供数据准备。

摘要由CSDN通过智能技术生成

#coding=utf-8
import os
#import os.path
import shutil  #Python文件复制相应模块
 
def GetFileNameAndExt(filename):
    (filepath,tempfilename) = os.path.split(filename);
    (shotname,extension) = os.path.splitext(tempfilename);
    return shotname,extension
 
'''source_dir='/home/.../VOC2007/ImageSets/Main/train'#在该文件得到文件名
img=os.listdir(source_dir)
print(img)'''
label_dir=r'/home/.../VOC2007/Annotations'  #与改路径拼接
annotion_dir='/home/.../teacher/train'  #拷贝到文件c
path = '/home/.../VOC2007/ImageSets/Main'
train = open(os.path.join(path,'train.txt'))
lines= train.readlines() 
#print (lines)
s=[]
for line in lines:
    line = line.strip()
    print (line)  #打印出文件名
    tempxmlname='%s.xml'%line
    print(tempxmlname)
    xmlname=os.path.join(label_dir,tempxmlname)
    print (xmlname)
    os.listdir(label_dir)
    shutil.copy(xm