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

有效

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

本申請實施例提供了一種圖像卷積處理方法和電子設備，獲取待卷積圖像和卷積核；判斷當前硬件平臺的硬件資源，是否滿足使用第一目標預設卷積分塊方式進行圖像卷積處理時對硬件資源的需求；若滿足，使用第一目標預設卷積分塊方式，對待卷積圖像和卷積核進行圖像卷積處理；若不滿足，根據預先設定的卷積核的個數與卷積分塊方式的對應關系，從除第一目標預設卷積分塊方式之外的其他預設卷積分塊方式中，選擇與卷積核的個數對應的第二目標預設卷積分塊方式，使用第二目標預設卷積分塊方式對待卷積圖像和卷積核進行圖像卷積處理。本申請實施例中，在多種卷積分塊方式中選擇目標卷積分塊方式，提高了圖像卷積處理的效率。

技術領域

本申請涉及圖像處理技術領域，特別是涉及一種圖像卷積處理方法和電子設備。

背景技術

在圖像處理技術領域，經常需要對圖像中的數據進行卷積處理，從而達到特征提取的目的。卷積處理的過程為：分別將待卷積圖像和卷積核映射為兩個矩陣：待卷積圖像矩陣和卷積核矩陣，從而將待卷積圖像與卷積核之間的卷積運算，轉化為上述待卷積圖像矩陣和卷積核矩陣的乘法運算，并對矩陣乘法運算得到的乘積矩陣進行反向映射，從而得到圖像卷積結果。

目前，一般采用分塊卷積的方法獲得卷積結果，具體的：按照預先設定的劃分方法，將待卷積圖像矩陣和卷積核矩陣分別劃分為多個待卷積圖像分塊矩陣和多個卷積核分塊矩陣，然后分別計算各待卷積圖像分塊矩陣和各卷積核分塊矩陣的乘積，得到多個分塊乘積矩陣，并對上述多個分塊乘積矩陣進行累加運算，得到乘積矩陣，再對乘積矩陣進行反向映射，從而得到圖像卷積結果。

上述分塊卷積方法中，對于任意尺寸的待卷積圖像或者卷積核，分塊方法是固定的，即：對于任意尺寸的待卷積圖像或者卷積核，分塊得到的待卷積圖像分塊矩陣的尺寸均是相同的，卷積核分塊矩陣的尺寸均是相同的，相應的，分塊乘積矩陣的尺寸也是相同的，而在實際處理過程中，針對部分尺寸的待卷積圖像或者卷積核，采用上述分塊卷積方法，處理效率較低。

發明內容

本申請實施例的目的在于提供一種圖像卷積處理方法、裝置、電子設備及存儲介質，以提高圖像卷積處理的效率。具體技術方案如下：

第一方面，本申請實施例提供了一種圖像卷積處理方法，包括：

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

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

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

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

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

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

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

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

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