[發明專利]基于改進SSD神經網絡的統計樹木數量的方法及系統

在審

• 申請號: 202010635369.5
• 申請日: 2020-07-03
• 公開（公告）號: CN111860623A
• 公開（公告）日: 2020-10-30
• 發明（設計）人: 韓巧玲;劉雷;趙燕東;趙玥;席本野;宋美慧;李晨曦
• 申請（專利權）人: 北京林業大學
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06T7/00;G06N3/04;G06N3/08
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 楊云云

1.一種基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，包括：

采集圖像，其中，所述圖像為航拍的環境圖像；

將所述圖像輸入預先訓練好的用于識別樹木的神經網絡模型中，以通過所述神經網絡模型識別所述圖像中的樹木，

其中，所述神經網絡模型包括一一對應地提取多個尺度的特征圖像的多個卷積層、以及一一對應地對多個尺度的特征圖進行特征融合的多個反卷積層，以由所述多個反卷積層獲取到融合后的新特征圖。

2.根據權利要求1所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，還包括對所述神經網絡模型進行訓練的步驟，具體為：

獲取多個圖像訓練樣本；

對多個圖像訓練樣本進行多尺度的特征圖映射；

對多尺度的特征圖進行先驗框匹配和數據增強訓練，得到訓練完成的用于識別樹木的所述神經網絡模型。

3.根據權利要求2所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，所述對多個圖像訓練樣本進行多尺度的特征圖映射，包括：

使用可提取多個尺度的特征圖的多個卷積層提取所述多個尺度的特征圖，并通過與對應的反卷積層的前后端融合，得到融合后的新特征圖。

4.根據權利要求2所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，所述對多尺度的特征圖進行先驗框匹配和數據增強訓練，得到訓練完成的用于識別樹木的所述神經網絡模型，包括：

基于多個尺度的特征圖，設置尺度或者長寬比不同的先驗框；

通過水平翻轉、裁剪、放大和縮小中的一個或多個方式對多尺度的特征圖進行數據增強。

5.根據權利要求2所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，所述對所述神經網絡模型進行訓練，包括：

對于多個圖像訓練樣本中每個真實目標，找到與其交并比IOU最大的先驗框；

通過所述先驗框與對應的真實目標進行匹配；

對剩余的未匹配的先驗框，如果真實目標的交并比IOU大于預設值，則通過所述先驗框與對應的真實目標進行匹配；

對負樣本進行抽樣，抽樣時按照置信度誤差進行降序排列，并選取誤差的較大的top-k作為訓練的負樣本。

6.根據權利要求2所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，所述通過所述神經網絡模型識別所述圖像中的樹木時，所述多尺度的特征圖的尺度越大，用于檢測的目標越小。

7.根據權利要求2所述的基于改進SSD神經網絡的統計樹木數量的方法，其特征在于，在所述神經網絡模型中，通過設置的損失函數進行多個圖像訓練樣本的分類。

8.一種基于改進SSD神經網絡的統計樹木數量的系統，其特征在于，包括：

采集模塊，用于采集圖像，其中，所述圖像為航拍的環境圖像；

分類模塊，用于將所述圖像輸入預先訓練好的用于識別樹木的神經網絡模型中，以通過所述神經網絡模型識別所述圖像中的樹木，其中，所述神經網絡模型包括一一對應地提取多個尺度的特征圖像的多個卷積層、以及一一對應地對多個尺度的特征圖進行特征融合的多個反卷積層，以由所述多個反卷積層獲取到融合后的新特征圖。

9.一種電子設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其特征在于，所述處理器執行所述計算機程序時實現根據權利要求1至7任一所述的基于改進SSD神經網絡的統計樹木數量的方法。

10.一種非暫態計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，該計算機程序被處理器執行時實現根據權利要求1至7任一所述的基于改進SSD神經網絡的統計樹木數量的方法。