数据集制作（一）

１．将图片按顺序重命名：

import os
path = "/home/customer/Desktop/voc/test2/"
filelist = os.listdir(path)
count=0
for file in filelist:
    print(file)
for file in filelist:                       #遍历所有文件
    Olddir=os.path.join(path,file)          #原来的文件路径
    if os.path.isdir(Olddir):               #如果是文件夹则跳过
        continue
    filename=os.path.splitext(file)[0]        #文件名
    filetype=os.path.splitext(file)[1]     #文件扩展名
    Newdir=os.path.join(path,str(count).zfill(6)+filetype)  #用字符串函数zfill 以0补全所需位数
    os.rename(Olddir,Newdir)                         #重命名
    count+=1
#命名改成000000.jpg格式。

２．将图片名放进txt

import os 
def generate(dir): 
    files = os.listdir(dir) 
    files.sort() 
    print ('****************') 
    print ('input :',dir )
    print ('start...')
    listText = open(dir+'/'+'test2.txt','w') 
    for file in files: 
        fileType = os.path.split(file) 
        if fileType[1] == '.txt': 
            continue 
        name = file + ' '  +'\n' 
        listText.write(name) 
    listText.close() 
    print ('down!') 
    print ('****************')
                     
if __name__ == '__main__': 
    generate('/home/customer/Desktop/voc/test2')