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

_一点点_

于 2019-04-25 15:52:15 发布

阅读量1.6k

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/weixin_41991401/article/details/89517694

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

#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(xmlname,annotion_dir)
    #os.listdir(xmlname)
 
'''   
##1.将指定A目录下的文件名取出,并将文件名文本和文件后缀拆分出来
img=os.listdir(source_dir)#得到文件夹下所有文件名称