[發明專利]一種圖像卷積處理方法和電子設備

有效

• 申請號: 202010196706.5
• 申請日: 2020-03-19
• 公開（公告）號: CN113495786B
• 公開（公告）日: 2023-10-13
• 發明（設計）人: 屠震元;葉挺群
• 申請（專利權）人: 杭州?？低晹底旨夹g股份有限公司
• 主分類號: G06F9/50
• 分類號: G06F9/50;G06V10/96;G06V10/94;G06V10/44
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 項京;丁蕓

1.一種圖像卷積處理方法，其特征在于，包括：

獲取待卷積圖像和卷積核；

判斷當前硬件平臺的硬件資源，是否滿足使用第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求；其中，所述第一目標預設卷積分塊方式為多種預設卷積分塊方式中，使得分塊尺寸最小的卷積分塊方式；

若滿足，則使用所述第一目標預設卷積分塊方式，對所述待卷積圖像和卷積核進行圖像卷積處理，得到圖像卷積結果；

若不滿足，則根據預先設定的卷積核的個數與卷積分塊方式的對應關系，從除所述第一目標預設卷積分塊方式之外的其他預設卷積分塊方式中，選擇與所述卷積核的個數對應的第二目標預設卷積分塊方式，使用所述第二目標預設卷積分塊方式對所述待卷積圖像和卷積核進行圖像卷積處理，得到圖像卷積結果。

2.根據權利要求1所述的方法，其特征在于，

在所述判斷當前硬件平臺的硬件資源，是否滿足使用第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求的步驟之前，還包括：

將所述待卷積圖像和卷積核分別映射為待卷積圖像矩陣和卷積核矩陣；

所述判斷當前硬件平臺的硬件資源，是否滿足使用第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求的步驟，包括：

計算使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，需要的線程塊的第一線程塊數量；其中，一個線程塊用于對所述待卷積圖像矩陣中的一個分塊矩陣和所述卷積核矩陣中的一個分塊矩陣進行矩陣乘法運算；

計算使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，基于當前硬件平臺的硬件資源所能夠運行的線程塊的第二線程塊數量；

如果所述第二線程塊數量大于或者等于所述第一線程塊數量，則確定當前硬件平臺的硬件資源，滿足使用所述第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求；

如果所述第二線程塊數量小于所述第一線程塊數量，則確定當前硬件平臺的硬件資源，不滿足使用所述第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求。

3.根據權利要求2所述的方法，其特征在于，

所述計算使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，需要的線程塊的第一線程塊數量的步驟，包括：

基于使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，所述待卷積圖像矩陣的一個分塊矩陣的尺寸和所述卷積核矩陣的一個分塊矩陣的尺寸，確定第一分塊乘積矩陣的尺寸；所述第一分塊乘積矩陣為使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，所述待卷積圖像矩陣的一個分塊矩陣和所述卷積核矩陣的一個分塊矩陣的乘積；

對所述待卷積圖像矩陣的行數與所述第一分塊乘積矩陣的行數的比值進行向上取整操作，得到第一比值；

對所述卷積核矩陣的列數與第一分塊乘積矩陣的列數的比值進行向上取整操作，得到第二比值；

將所述第一比值和所述第二比值的乘積，確定為使用所述第一目標預設卷積分塊方式進行圖像卷積處理時，需要的線程塊的第一線程塊數量。

4.根據權利要求2或3所述的方法，其特征在于，

所述使用所述第一目標預設卷積分塊方式，對所述待卷積圖像和卷積核進行圖像卷積處理，得到圖像卷積結果的步驟，包括：

按照所述第一目標預設卷積分塊方式中的劃分方法，將所述待卷積圖像矩陣和所述卷積核矩陣分別劃分為多個第一待卷積圖像分塊矩陣和多個第一卷積核分塊矩陣；

分別計算所述多個第一待卷積圖像分塊矩陣和所述多個第一卷積核分塊矩陣的乘積，得到多個所述第一分塊乘積矩陣；

對所述多個第一分塊乘積矩陣進行累加運算，得到乘積矩陣；

對所述乘積矩陣進行反向映射處理，得到圖像卷積結果；

所述使用所述第二目標預設卷積分塊方式對所述待卷積圖像和卷積核進行圖像卷積處理，得到圖像卷積結果的步驟，包括：

按照所述第二目標預設卷積分塊方式中的劃分方法，將所述待卷積圖像矩陣和所述卷積核矩陣分別劃分為多個第二待卷積圖像分塊矩陣和多個第二卷積核分塊矩陣；

分別計算所述多個第二待卷積圖像分塊矩陣和所述多個第二卷積核分塊矩陣的乘積，得到多個第二分塊乘積矩陣；

對所述多個第二分塊乘積矩陣進行累加運算，得到乘積矩陣；

對所述乘積矩陣進行反向映射處理，得到圖像卷積結果。