[發明專利]執行神經網絡運算的指令生成方法、圖像處理方法及裝置

在審

• 申請號: 202011498886.9
• 申請日: 2020-12-16
• 公開（公告）號: CN112559042A
• 公開（公告）日: 2021-03-26
• 發明（設計）人: 賀浩千;王振江;譚洪賀
• 申請（專利權）人: 北京地平線信息技術有限公司
• 主分類號: G06F9/30
• 分類號: G06F9/30;G06N3/10;G06N3/04;G06T1/40
• 代理公司: 北京思源智匯知識產權代理有限公司 11657
• 代理人: 毛麗琴

1.一種執行神經網絡運算的指令生成方法，包括：

基于神經網絡中的第一中間層組，確定至少一個中間層集合；所述第一中間層組與所述神經網絡中的第一輸入層組對應，且用于輸入所述第一輸入層組的圖像幀的數量不固定；

分別確定所述至少一個中間層集合中的每一中間層集合的中間層編號信息和中間層數量；

基于每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令。

2.根據權利要求1所述的方法，其中，所述基于每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令，包括：

基于所述第一輸入層組的組標識，以及每一中間層集合的中間層編號信息和中間層數量，分別確定每一中間層集合對應的三元組；

分別給每一中間層集合對應的三元組分配寄存器；

基于給每一中間層集合對應的三元組分配的寄存器，以及每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令。

3.根據權利要求1所述的方法，其中，所述基于每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令，包括：

分別給每一中間層集合分配單個中間層的輸入存儲空間和輸出存儲空間；

基于給每一中間層集合分配的單個中間層的輸入存儲空間和輸出存儲空間，以及每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令。

4.一種圖像處理方法，包括：

基于目標幀數量，以及根據神經網絡中的第一中間層組確定的至少一個中間層集合中的每一中間層集合的中間層編號信息和中間層數量，分別給每一中間層集合確定待執行的運算次數；所述第一中間層組與所述神經網絡中的第一輸入層組對應，用于輸入所述第一輸入層組的圖像幀的數量不固定，且所確定的待執行的運算次數的總和與所述目標幀數量在數值上相等；

通過每一中間層集合，按照所確定的待執行的運算次數，對所述目標幀數量的圖像幀執行上述權利要求1生成的用于神經網絡運算的指令，以對所述目標幀數量的圖像幀進行運算處理。

5.根據權利要求4所述的方法，其中，所述第一中間層組中的中間層按照預設編號規則進行編號；

所述基于目標幀數量，以及根據神經網絡中的第一中間層組確定的至少一個中間層集合中的每一中間層集合的中間層編號信息和中間層數量，分別給每一中間層集合確定待執行的運算次數，包括：

基于目標幀數量，根據神經網絡中的第一中間層組確定的至少一個中間層集合中的每一中間層集合的中間層編號信息和中間層數量，以及與所述預設編號規則匹配的運算次數確定方式，分別給每一中間層集合確定待執行的運算次數。

6.根據權利要求5所述的方法，其中，所述通過每一中間層集合，按照所確定的待執行的運算次數，對所述目標幀數量的圖像幀執行上述權利要求1生成的用于神經網絡運算的指令，包括：

將給每一中間層集合對應的三元組分配的寄存器中的存儲值設置為給本中間層集合確定的待執行的運算次數的數值；

通過每一中間層集合，按照給每一中間層集合對應的三元組分配的寄存器中設置后的存儲值，對所述目標幀數量的圖像幀執行上述權利要求2生成的用于神經網絡運算的指令。

7.一種執行神經網絡運算的指令生成裝置，包括：

第一確定模塊，用于基于神經網絡中的第一中間層組，確定至少一個中間層集合；所述第一中間層組與所述神經網絡中的第一輸入層組對應，且用于輸入所述第一輸入層組的圖像幀的數量不固定；

第二確定模塊，用于分別確定所述第一確定模塊確定出的所述至少一個中間層集合中的每一中間層集合的中間層編號信息和中間層數量；

生成模塊，用于基于所述第二確定模塊確定出的每一中間層集合的中間層編號信息和中間層數量，生成用于執行所述神經網絡運算的指令。