#按照文件名设置标签并将其插入文件名中
import os
import shutil
#获取目标文件夹的路径
a= r'C:UsersyyDesktop ianchi大赛guangdong_round1_train2_20180910无瑕疵样本'
#b=r'C:UsersyyDesktop ianchi大赛 rian'
tag_dict={'不导电':'t00','凸粉':'t01','擦花':'t02','桔皮':'t03','横条压凹':'t04','涂层开裂':'t05','漏底':'t06',
'碰伤':'t07','脏点':'t08','起坑':'t09','不导电负样本':'f00','凸粉负样本':'f01','擦花负样本':'f02','桔皮负样本':'f03',
'横条压凹负样本':'f04','涂层开裂负样本':'f05','漏底负样本':'f06','碰伤负样本':'f07','脏点负样本':'f08','起坑负样本':'f09'}
#获取当前文件夹中的文件名称列表
filenames=os.listdir(a)
for i in filenames:
if i!='其他':
print(i)
imgdir=a+'\'+i
imgnames=os.listdir(imgdir)
for imgname in imgnames:
#print(imgdir+'\'+imgname)
imgdirname=imgdir+'\'+imgname
site=imgdirname.find('.')
#imgname[0:len(i)]=tag_dict[i]
os.rename(imgdirname,imgdirname[0:site]+tag_dict[i+'负样本']+'.jpg')
#print(imgdirname[0:site]+tag_dict[i]+'.jpg')
#f=open(b+'\'+filename,'a')
''' filepath=a+'\'+filename
shutil.copyfile(filepath,b+str(i)+'\'+filename+'.jpg')'''
#遍历单个文件,读取行数