1. 程式人生 > >深度學習大規模道路網路自動提取

深度學習大規模道路網路自動提取

       好久沒有更新了,近期做了一個工作,就是用深度學習方法來自動提取高解析度遙感影像上的道路網路,16年的時候都是用的什麼SVM,什麼增強線性指數等

手工設計的線性目標增強特徵,然後去提取道路,那些個東西說白了,沒啥用,在一個小區域可能做的有那麼點效果,但是換一個場景,立馬就沒了,沒有任何普適性

可言,之前用深度學習方法做過高解析度影像變化監測工作,這一次,我重新把網路進行了設計,用於提取道路網路,最終生成道路中心,並進行大規模場景實驗,結果

還是非常不錯的,個人覺得在一定程度上可以減少人工提取道路網路的工作量。如需要程式碼:Email:[email protected],qq:1044625113,新增好友時,請備註:道路網路自動提取!

       好了不多說了,介紹下道路網路提取的部分工作,首先是樣本部分,樣本採用開源的道路網路資料和標籤(應該是國外某個地區),大家可以到開源資料網站下載,

訓練集大概是6000多張1024*1024的三通道高解析度影像,看下面:

                                           圖1 樣本和標籤

     其實,客觀的說,這個樣本做的確實不夠好,為什麼呢,因為有一些道路該畫出來的,沒有畫出來,比如說這個樣本上的中間那條路,就沒有畫出來,但是總的來說,在深度學習裡面,

樣本是燃料,有總比沒有要好,是吧!接下來就輸入到我設計的網路中來,總體來說,參考了Deeplabv3+的思想,使用多尺度特徵進行提取,但是全部採用分離卷積替換resnet50層,這樣

有一個好處,學習速度快,而且輕量,比Deeplab輕了5倍,更為關鍵的是,一張卡可以扔進更多的樣本。一張RTX2080ti可以扔進16個1024*1024的樣本,而一般的Deeplabv3v 只能扔進4個,

多了就out of memory了,可能在學術上意義不大,但是在工程上意義巨大,因為工頭往往不會有那麼多時間給你訓練!!!!!(開個玩笑)

    模型訓練完畢,測試結果程式碼如下所示:

function resultBlock = testDLmodel(imgBlock, inputSize, net)
% we can use this function to process big image

TestImg = imgBlock(:,:,1:3);                                            % google 地圖下載的是四個通道

patchSize = inputSize;                                                 % 按照訓練影象塊的大小去分塊預測
Result = segmentImage(TestImg, net, patchSize);
Result = im2bw(mat2gray(Result), 0.1);

line = extractionLine(Result);                                         % 提取中心線

se = strel('disk',2);
line = imdilate(line, se);
resultBlock = labeloverlay(mat2gray(TestImg),line,'Transparency',0);          % 疊加到原始影像上

fprintf(['a block has been finished...\n']);

% imwrite(mat2gray(line),[path(1:end-4),'_LiteSeg_out_Line.tif']);
% imwrite(mat2gray(Result),[path(1:end-4),'_LiteSeg_out.tif']);

end

  由於影像實在太大,我用了分塊處理的方式進行解決,這樣簡單而又方便的解決IO,記憶體不足的問題了,這個比較工程手段了,其實我個人學術上也做了一點工作,

我發現學院派有一個很大的缺點,就是小區域做個實驗就完事了,但是真正的工程不是這樣的,小區域實驗成功後,應當大規模應用起來,所以我的部落格都是一些實際而

又有用的工作,比如下面這個分塊處理大影像問題:

%% read test image
load net_liteSegModel.mat
inputSize = [4096, 4096];


%% block process to extract road line
fun = @(block_struct) testDLmodel(block_struct.data, inputSize, net);        % 呼叫分塊處理超大影像
blockproc('嶽麓區.tif', inputSize, fun, 'Destination', '嶽麓區中心線.tif');

     在開源資料訓練完畢後,我直接爬了google資料進行測試,地區選擇國內湖南地區,大小是4萬*3萬畫素,總共是14個g,跑完大概花了30分鐘,如果換成固態硬碟會快的多,如果並行的話,

控制在10分鐘以內應該問題不大,主要時間花在IO上面了,窮人只能用機械硬碟,唉。。。。。好了,直接看結果吧!

 

                                                                            圖2 道路提取結果

      圖中藍色區域為道路中心線,大家可能看的不是非常清晰,下面我放大幾個區域:

                       

                                                                         圖3 農田部分

                        

                                                                        圖4 山區公路部分  

                        

                                                                圖5 山區盤山泥濘小路

                        

                                                             圖6 丘陵地帶道路提取結果

     從結果來看,確實是不錯的,但是仍然存在部分道路網路斷裂,彎曲等等,這些可以通過後續優化來改進,不是特別難的問題了,好吧,如果大家有需要的程式碼的,請聯絡我吧。

     

&n