[發(fā)明專利]神經(jīng)網(wǎng)絡中的卷積計算方法和電子設備

有效

• 申請?zhí)枺?/td>: 201711361938.6
• 申請日: 2017-12-18
• 公開（公告）號: CN108108811B
• 公開（公告）日: 2021-07-30
• 發(fā)明（設計）人: 陳亮;黃暢;凌坤;李建軍;李德林;羅恒
• 申請（專利權）人: 南京地平線機器人技術有限公司
• 主分類號: G06N3/063
• 分類號: G06N3/063;G06N3/04
• 代理公司: 北京市正見永申律師事務所 11497
• 代理人: 黃小臨;王懷章

1.一種神經(jīng)網(wǎng)絡中的卷積計算方法，包括：

根據(jù)輸入特征圖、逐層卷積核、和逐點卷積核進行逐層卷積計算和逐點卷積計算，以得到所述輸入特征圖的第一預定數(shù)目p個點在所有逐點卷積輸出通道上的輸出特征值，所述第一預定數(shù)目p是根據(jù)所述輸入特征圖的寬高維度和輸入通道數(shù)維度中的至少一個確定的；以及

重復上述操作，以得到所有點在所有逐點卷積輸出通道上的輸出特征值；

其中，根據(jù)輸入特征圖、逐層卷積核、和逐點卷積核進行逐層卷積計算和逐點卷積計算，以得到所述輸入特征圖的第一預定數(shù)目p個點在所有逐點卷積輸出通道上的輸出特征值包括：

根據(jù)輸入特征圖和逐層卷積核進行逐層卷積計算，以得到第一預定數(shù)目p個點在所有逐層卷積輸出通道上的中間特征值；以及

根據(jù)所述第一預定數(shù)目p個點在所有逐層卷積輸出通道上的中間特征值和逐點卷積核進行逐點卷積計算，以得到第一預定數(shù)目p個點在所有逐點卷積輸出通道上的輸出特征值。

2.如權利要求1所述的方法，其中，根據(jù)輸入特征圖和逐層卷積核進行逐層卷積計算，以得到第一預定數(shù)目p個點在所有逐層卷積輸出通道上的中間特征值包括：

根據(jù)所述輸入特征圖和所述逐層卷積核進行逐層卷積計算，以得到所述第一預定數(shù)目p個點在第二預定數(shù)目m個逐層卷積輸出通道上的中間特征值；以及

重復上述操作，根據(jù)所述輸入特征圖和所述逐層卷積核進行逐層卷積計算，以得到所述第一預定數(shù)目p個點在接下來的第二預定數(shù)目m個逐層卷積輸出通道上的中間特征值，并相應地執(zhí)行后續(xù)操作，直到得到所述第一預定數(shù)目p個點在所有逐層卷積輸出通道上的中間特征值為止。

3.如權利要求2所述的方法，其中，所述第二預定數(shù)目m是根據(jù)逐層卷積計算單元的數(shù)目和所述第一預定數(shù)目p來確定的。

4.如權利要求2所述的方法，其中，根據(jù)所述輸入特征圖和所述逐層卷積核進行逐層卷積計算，以得到所述第一預定數(shù)目p個點在第二預定數(shù)目m個逐層卷積輸出通道上的中間特征值包括：

從所述輸入特征圖中讀取第一預定數(shù)目p組點在第二預定數(shù)目m個輸入通道上的輸入特征值；

與所述第一預定數(shù)目p組點在所述第二預定數(shù)目m個輸入通道上的輸入特征值對應地，讀取對應的第二預定數(shù)目m個逐層卷積核中的權重值；以及

分別對所述第一預定數(shù)目p組點在所述第二預定數(shù)目m個輸入通道上的輸入特征值與對應的第二預定數(shù)目m個逐層卷積核中的權重值進行逐層卷積計算，以得到與所述第一預定數(shù)目p組點分別對應的所述第一預定數(shù)目p個點在所述第二預定數(shù)目m個逐層卷積輸出通道上的中間特征值。

5.如權利要求4所述的方法，其中，從所述輸入特征圖中讀取第一預定數(shù)目p組點在第二預定數(shù)目m個輸入通道上的輸入特征值包括：

以逐層卷積核中的權重值的寬高為一組點的寬高、以所述逐層卷積計算的步長為兩組點之間的讀取步長，從所述輸入特征圖中讀取所述第一預定數(shù)目p組點在所述第二預定數(shù)目m個輸入通道上的輸入特征值。