根据文件夹更改样本文件名小程序.py

Posted 2021-01-23 Yangami

#按照文件名设置标签并将其插入文件名中
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‘)‘‘‘
        #遍历单个文件，读取行数