pascal voc 2012提取某一类的图片

# 读取 F:\数据集\Pascal voc 2012\VOCtrainval_11-May-2012\VOCdevkit\VOC2012_2\Annotations中的所有文件名
# 得到所有有牛的图片的名字和牛图片数量
import os
import re

rootdir = 'F:\数据集\Pascal voc 2012\VOCtrainval_11-May-2012\VOCdevkit\VOC2012_2\Annotations'
outfile = 'F:\数据集\Pascal voc 2012\VOCtrainval_11-May-2012\VOCdevkit\VOC2012_2\AllCow.txt'   #写入的txt文件名
wildcard = ".xml"               #要读取的文件类型
file_write = open(outfile,'w')
cow_img_num = 0
lists = os.listdir(rootdir)     #列出文件夹下所有的目录与文件
print(str(len(lists)) + "个文件")
for name in lists:              # name是.xml的文件名
    # print(name)                 #对每个.xml文件,打开、读取文字
    path = 'F:\数据集\Pascal voc 2012\VOCtrainval_11-May-2012\VOCdevkit\VOC2012_2\Annotations\\' + name
    # print(path)               # path是.xml文件的路径
    file = open(path, 'r')      # 打开.xml文件
    contents = file.read()      # 读取.xml文件的内容
    # print(contents)           # .xml文件的内容输出
    t = re.search(r'cow', contents)
    if t:
        print(name)
        file_write.write(name + "\n")   # 将有牛的图片的名字放到txt文件
        cow_img_num = cow_img_num + 1
print('一共' + str(cow_img_num) + '张牛图片')

#########################################################################
# 读取 F:\数据集\Pascal voc 2012\VOCtrainval_11-May-2012\VOCdevkit\VOC2012_2\JPEGImages中的所有牛牛文件名
# 得到所有有牛的图片的名字和牛图片数量
import os