python 从txt文件查找含某字段的行,并将这些行名称的图像提取出来另存
将voc数据集中bottle_train.txt文件中的正图片提取出来另存到文件夹中
#-*- coding:UTF-8 -*- import re from PIL import Image f1 = open('/home/wei/MyPaperCode/VOC2012/ImageSets/Main/bottle_train.txt') f2 = open('bottle_train.txt', 'w+') for line in f1.readlines(): if re.findall(' 1', line):#查找“空格1”的行 f2.write(line)#将查找到的行写入f2 f1.close() f2.close() f3 = open('bottle_train.txt', 'r') for line2 in f3.readlines(): line3 = line2[:-4]#读取每行去掉后四位的数 im = Image.open('JPEGImages/{}.jpg'.format(line3))#打开该路径下line3记录的文件名 im.save('myDataSet/bottle/{}.jpg'.format(line3))#把文件夹中指定的文件名的图片另存到该路径下 f3.close()
将voc数据集中bottle_train.txt文件中的正图片的标注提取出来另存到文件夹中,此方法也可用于上面的图片另存
#-*-coding:utf-8 -*- import os import shutil f1 = open('bottle_train.txt', 'r') for line in f1.readlines(): line1 = line[:-4] f2 = ('/home/wei/MyPaperCode/VOC2012/Annotations/{}.xml'.format(line1)) shutil.copy(f2, 'myDataSet/Annotations/')