(三)根据txt文件批量提取图片
python代码根据txt文件提取图片
从数据集中提取测试集图片
# -*- coding: UTF-8 -*-
import sys
import re
import numpy as np
import shutil
data = []
for line in open("save.txt", "r"): # 设置文件对象并读取每一行文件
data.append(line)
for a in data:
#print(a)
line3=a[:-1]
#print(line3)
#line3=a[:-5] #读取每行去掉后四位的数,比如使用的格式为000001.jpg,即去掉.jpg
#print('line3', line3)
line4 = line3 + '.jpg'
print(line4)
oldname = r'train2017/{}'.format(line4)
#print('old', oldname)
newname = r'test/{}'.format(line4)
#print('new', newname)
shutil.move(oldname, newname) #将需要的文件从oldname移动到newname
#shutil.copyfile(oldname, newname) #将需要的文件从oldname复制到newname