1. 程式人生 > >根據文件夾更改樣本文件名小程序.py

根據文件夾更改樣本文件名小程序.py

str don ren user 根據 shutil 路徑 lena site

#按照文件名設置標簽並將其插入文件名中
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‘)‘‘‘
#遍歷單個文件,讀取行數

根據文件夾更改樣本文件名小程序.py