天天看點

SSD訓練自己的資料集(一): 制作VOC資料集

準備自己的圖檔,最好重命名為“000001.jpg”的格式,重命名Python代碼如下:

import os
path = "./123"
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
           

VOC下一共有五個檔案夾,其中隻用到前三個檔案夾,Annotations、JPEGImages和ImageSets

SSD訓練自己的資料集(一): 制作VOC資料集

建立以上三個檔案夾,再ImageSets檔案夾下在建立Main檔案夾,Main檔案夾下存放四個txt檔案

SSD訓練自己的資料集(一): 制作VOC資料集

其中test.txt儲存用于測試集的圖檔編号(如“00012

繼續閱讀