1. 程式人生 > >SSD訓練自己的資料集 Ubuntu16.04

SSD訓練自己的資料集 Ubuntu16.04

環境:Ubuntu16.04  cuda8.0 cudnn6

一、製作自己的資料集

https://github.com/whlook/VOCMaker

imgs存放所有的原始圖片



JPEGImage 存放所有的訓練集圖片。

(1)首先執行rename.py檔案



執行rename.py 指令碼來生成基本目錄並將圖片命名轉換成標準格式存放在JPEGImages

(2)開啟vs 2017

執行檔案並進行標註,

“O”完成標註並生成對應的xml檔案,對應在Annotations資料夾下。



(3)執行txt指令碼,


在Main資料夾下生成對應的xml檔案。



test.txt中儲存的是測試所用的所有樣本的名字,不過沒有後綴(下同),一般測試的樣本數量佔總資料集的50%

train.txt中儲存的是訓練所用的樣本名,樣本數量通常佔trainval的50%左右

trainval.txt中儲存的是訓練驗證樣本,是上面兩個的總和,一般數量佔總資料集的50%

val.txt中儲存的是驗證所用的樣本名,數量佔trainval的50%左右

擴充資料

(1)

# -*- codin: utf-8 -*-"""Spyder Editor
This is a temporary script file."""##t圖片旋轉90.180.270from PIL import Imagefor n in range (1,4): for m in range (1,82): im = Image.open(r
"E:/yogart/5.4/%06d.jpg" %m) im = im.rotate(90*n) im.save(r"E:\yogart\5.4trian/%06d-%d.jpg" % (m, n))

(2)keras