Yangami

导航

 

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

posted on 2018-11-29 08:43  Yangami  阅读(130)  评论(0编辑  收藏  举报