(1)写图片路径到txt中小栗子:
# -*- coding: utf-8 -*-
import glob
a=glob.glob('layers/*.jpg')
for i in range(len(a)):
a[i]=a[i].split('/')[-1]
f=file('list.txt','w')
for i in a:
f.write(i+'\n')
f.close()
(2)提取图片路径中的标签小栗子:
# -*- coding: utf-8 -*-
with open("/home/sll2/Desktop/train_s.txt","r")as f:
lines=f.readlines()
with open("/home/sll2/Desktop/train_s.txt","w+")as f_w:
for line in lines:
label=line.split('/')[-3]
#label=int(label)
line=line.strip('\n')
f_w.write(line+' '+label+'\n')
f.close()
f_w.close()