根據資料夾更改樣本檔名小程式.py
#按照檔名設定標籤並將其插入檔名中
import os
import shutil
#獲取目標資料夾的路徑
a= r'C:\Users\yy\Desktop\tianchi大賽\guangdong_round1_train2_20180910\無瑕疵樣本'
#b=r'C:\Users\yy\Desktop\tianchi大賽\trian'
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')'''
#遍歷單個檔案,讀取行數