yolo-windows檢測超大圖及封裝應用
ofollow,noindex">YOLO訓練自己的資料集的一些心得
專案開原始碼:LargeImageDetect-yolo-windows
由於專案需要,需要將yolo移植到windows下並進行封裝並自動測試,遇到的主要問題如下:
1.yolo測試的圖片一般不會太大,基本為468*468,或者512*512,專案中需要檢測的圖片較大,多為1W*1W起;
2.原始碼中為cmd命令列進行圖片測試,資料夾測試需要不斷輸入路徑;
3.原始碼體量很大,不適合移植。
解決方法:
1.訓練的模型輸入圖片size為512*512,因此採用滑窗演算法對超大圖進行裁剪存入資料夾,進行****_****特殊命名;
2.修改命令列測試為資料夾測試,每次測試一個資料夾,將1中裁剪後的圖片資料夾送入yolov3進行檢測,利用命名規則對檢測後的各部分圖片進行融合;
3.對原始碼封裝為lib進行移植
-----------------------------------------
命名規則如下:
0000_0000:表示裁剪的第一行第一列圖片,依次遞增,最大可以裁剪10000*10000張圖片,其中裁剪大小和步長可以修改