[發明專利]一種用于神經網絡的池化方法及裝置

在審

• 申請號: 202010039300.6
• 申請日: 2020-01-14
• 公開（公告）號: CN113191480A
• 公開（公告）日: 2021-07-30
• 發明（設計）人: 范坤;陳邁越
• 申請（專利權）人: 北京地平線機器人技術研發有限公司
• 主分類號: G06N3/04
• 分類號: G06N3/04;G06N3/08
• 代理公司: 北京嘉科知識產權代理事務所(特殊普通合伙) 11687
• 代理人: 楊波

1.一種用于神經網絡的池化方法，包括：

將圖像輸入神經網絡的卷積層，得到特征圖；

根據預設池化步長，獲取所述特征圖在預設數據窗口下的矩陣元素；

根據所述矩陣元素中的噪聲元素，確定所述矩陣元素的第一均值集合；

根據所述矩陣元素的第一均值集合，確定所述矩陣元素的池化輸出值。

2.根據權利要求1所述的方法，其中，所述根據所述矩陣元素中的噪聲元素，確定所述矩陣元素的第一均值集合，包括：

計算所述矩陣元素中噪聲元素之外的元素的第一均值；

根據所述第一均值，確定所述矩陣元素的第一均值集合。

3.根據權利要求2所述的方法，所述根據所述第一均值，確定所述矩陣元素的第一均值集合，包括：

確定所述矩陣元素中任意兩個相鄰元素之間的差值，所述兩個相鄰元素滿足預設條件；

根據各個所述差值，確定第二均值集合；

根據所述第一均值和第二均值集合，確定所述矩陣元素的第一均值集合。

4.根據權利要求3所述的方法，其中，所述根據各個所述差值，確定第二均值集合，包括：

確定各個所述差值分別對應的絕對值；

將各個所述差值分別對應的絕對值的均值確定為第二均值；

根據所述第二均值，確定第二均值集合。

5.根據權利要求4所述的方法，所述兩個相鄰元素滿足預設條件包括：所述兩個相鄰元素為所述矩陣元素中的行元素、列元素和對角線元素中的任意一項；

所述第二均值集合包括：所述矩陣元素中的行元素、列元素和對角線元素中的任意一項或多項對應的所述第二均值。

6.根據權利要求1所述的方法，其中，所述根據所述矩陣元素的第一均值集合，確定所述矩陣元素的池化輸出值，包括：

獲取所述第一均值集合中每個均值各自對應的權重值；

根據所述第一均值集合中每個均值各自對應的權重值，對所述第一均值集合中的均值進行加權求和，確定所述矩陣元素的池化輸出值。

7.根據權利要求1至6中任一項所述的方法，其中，所述噪聲元素包括所述矩陣元素中的最大值和/或最小值。

8.一種用于神經網絡的池化裝置，包括：

特征提取模塊，用于將圖像輸入神經網絡的卷積層，得到特征圖；

獲取模塊，用于根據預設池化步長，獲取所述特征圖在預設數據窗口下的矩陣元素；

計算模塊，用于根據所述矩陣元素中的噪聲元素，確定所述矩陣元素的第一均值集合；

輸出模塊，用于根據所述矩陣元素的第一均值集合，確定所述矩陣元素的池化輸出值。

9.一種計算機可讀存儲介質，所述存儲介質存儲有計算機程序，所述計算機程序用于執行上述權利要求1-7任一所述的用于神經網絡的池化方法。

10.一種電子設備，所述電子設備包括：

處理器；

用于存儲所述處理器可執行指令的存儲器；

所述處理器，用于從所述存儲器中讀取所述可執行指令，并執行所述指令以實現上述權利要求1-7任一所述的用于神經網絡的池化方法。