[發明專利]一種卷積神經網絡的數據調度方法、系統及計算機設備

有效

• 申請號: 201611205487.2
• 申請日: 2016-12-23
• 公開（公告）號: CN106874219B
• 公開（公告）日: 2018-11-02
• 發明（設計）人: 蔣文
• 申請（專利權）人: 深圳云天勵飛技術有限公司
• 主分類號: G06F12/0871
• 分類號: G06F12/0871;G06N3/06
• 代理公司: 廣州三環專利商標代理有限公司 44202
• 代理人: 郝傳鑫;熊永強

1.一種卷積神經網絡的數據調度方法，其特征在于，包括：

將圖像數據分成N個目標圖像數據，所述N為大于1的整數，所述N個目標圖像數據中包含第一目標圖像數據和第二目標圖像數據，所述第一目標圖像數據和所述第二目標圖像數據為相鄰的所述目標圖像數據；

將所述第一目標圖像數據載入第一數據緩沖模塊，計算單元讀取所述第一數據緩沖模塊存儲的所述第一目標圖像數據后進行卷積計算；在所述計算單元讀取所述第一數據緩沖模塊存儲的所述第一目標圖像數據后進行卷積計算的過程中，將所述第二目標圖像數據載入第二數據緩沖模塊；所述計算單元讀取所述第二數據緩沖模塊中的所述第二目標圖像數據后進行卷積計算；所述第一數據緩沖模塊和所述第二數據緩沖模塊交替執行載入操作；

所述計算單元得到計算結果后將所述計算結果上傳到外部存儲；

在所述將所述圖像數據分成N個所述目標圖像數據之前，所述方法還包括：

根據緩存當前可用的存儲空間以及計算單元的數量，確定第一數據緩沖模塊和第二數據緩沖模塊可分配的最大存儲空間，所述第一數據緩沖模塊和所述第二數據緩沖模塊可分配的最大存儲空間相同；

所述將圖像數據分成N個目標圖像數據包括：

確定若將所述圖像數據分成所述N個數據量相等的所述目標圖像數據后，所述第一目標圖像數據所需的存儲空間是否小于或等于所述第一數據緩沖模塊可分配的最大存儲空間；

若是，將所述圖像數據分成所述N個所述目標圖像數據。

2.根據權利要求1所述方法，所述將圖像數據分成N個目標圖像數據包括：

將所述圖像數據分成所述N個所需存儲空間相等的所述目標圖像數據；

所述將所述第一目標圖像數據載入第一數據緩沖模塊包括：

為所述第一數據緩沖模塊和第二數據緩沖模塊分配與所述目標圖像數據所需存儲空間相等的存儲空間，將所述第一目標圖像數據載入所述第一數據緩沖模塊。

3.根據權利要求1或2所述方法，所述計算單元得到計算結果后將所述計算結果上傳到外部存儲包括：

所述計算單元讀取所述第一數據緩沖模塊存儲的所述第一目標圖像數據后進行卷積計算，得到所述計算結果后將所述計算結果上傳到所述外部存儲；

或者，所述計算單元將計算出的計算結果存儲到緩存中，若所述緩存中存儲的所述計算結果的數據量達到預設條件，則上傳到所述外部存儲。

4.根據權利要求1所述方法，所述計算單元讀取所述第一數據緩沖模塊存儲的所述第一目標圖像數后進行卷積計算包括：

所述計算單元從緩存中讀取所述圖像數據的權重參數以及讀取所述第一數據緩沖模塊存儲的所述第一目標圖像數據后，進行卷積計算，得到計算結果，將所述計算結果存儲到所述緩存。