1. 程式人生 > 其它 >【ENVI深度學習】尋找圖片中的特定符號

【ENVI深度學習】尋找圖片中的特定符號

1 概述

圖片是一張掃描件,3波段影象。我們的任務是定位/計數一個特定的符號。目標符號的顏色和地圖上的許多其他特徵類似,所以我們不能僅僅依靠光譜(RGB值)來處理。

要尋找的符號

我們需要訓練一個深度學習模型,然後進行目標提取。

選擇訓練樣本

首先從原圖裁剪出一小塊影象作用樣本選擇和模型訓練的輸入影象。然後選擇訓練樣本,俗稱打標籤。在ENVI中可以方便的利用ROI工具打標籤。如下圖所示,利用點和多段線型別的ROI進行目標區域繪製。

如下圖是選擇的部分樣本ROI

輸出標籤柵格

工具為:/Deep Learning/Build Label Raster from ROI

 初始化模型

需要設定切片大小、波段數。ENVI深度學習的優勢在於可以處理多波段影象(多於3波段)。

訓練模型

深度學習訓練過程就是一個調參的過程。

 目標提取

/Deep Learning/TensorFlow Mask Classification

 結果檢視與後處理

輸出結果為一張灰度圖,DN值代表了該畫素屬於目標的概率。利用ENVI密度分割工具可以快去提取目標,如下為柵格結果:

可以將其轉換為向量結果