SSD訓練自己的資料集 Ubuntu16.04
阿新 • • 發佈:2019-01-09
環境: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 EditorThis 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
(2)keras