[發(fā)明專利]一種卷積處理方法、裝置、計算機設備及存儲介質

在審

• 申請?zhí)枺?/td>: 202110602237.7
• 申請日: 2021-05-31
• 公開（公告）號: CN113327217A
• 公開（公告）日: 2021-08-31
• 發(fā)明（設計）人: 蔣程;胡英俊;曹雨;勾志宏
• 申請（專利權）人: 上海陣量智能科技有限公司
• 主分類號: G06T5/50
• 分類號: G06T5/50
• 代理公司: 北京中知恒瑞知識產權代理事務所(普通合伙) 11889
• 代理人: 吳迪

本公開提供了一種卷積處理方法、裝置、計算機設備及存儲介質，其中，該方法包括：獲取待處理圖像數(shù)據(jù)，并將所述待處理圖像數(shù)據(jù)分為至少一組子圖像數(shù)據(jù)；針對所述至少一組子圖像數(shù)據(jù)中的每組子圖像數(shù)據(jù)，根據(jù)該組子圖像數(shù)據(jù)中各個子圖像分別對應的卷積操作數(shù)，生成該組子圖像數(shù)據(jù)對應的卷積核；利用該組子圖像數(shù)據(jù)對應的卷積核，對該組子圖像數(shù)據(jù)進行卷積處理，得到該組子圖像數(shù)據(jù)對應的輸出結果數(shù)據(jù)。本公開實施例實現(xiàn)利用AI芯片對深度可分卷積depthwise卷積進行處理。

技術領域

本公開涉及數(shù)據(jù)處理技術領域，具體而言，涉及一種卷積處理方法、裝置、計算機設備及存儲介質。

背景技術

卷積操作廣泛應用于深度學習、圖像處理等領域。其中，普通卷積、以及深度可分depthwise卷積應用較為廣泛，普通卷積一般通過人工智能(Artificial Intelligence，AI)芯片進行處理，但是因為普通卷積和depthwise卷積的算法原理不同，但AI芯片通常是基于有限的算子類型設計的，對于某些AI芯片而言，其僅能夠對普通卷積進行處理；造成depthwise卷積在AI芯片中處理受限。

發(fā)明內容

本公開實施例至少提供一種卷積處理方法、裝置、計算機設備以及存儲介質。

第一方面，本公開實施例提供了一種卷積處理方法，包括：獲取待處理圖像數(shù)據(jù)，并將所述待處理圖像數(shù)據(jù)分為至少一組子圖像數(shù)據(jù)；針對所述至少一組子圖像數(shù)據(jù)中的每組子圖像數(shù)據(jù)，根據(jù)該組子圖像數(shù)據(jù)中各個子圖像分別對應的卷積操作數(shù)，生成該組子圖像數(shù)據(jù)對應的卷積核；利用該組子圖像數(shù)據(jù)對應的卷積核，對該組子圖像數(shù)據(jù)進行卷積處理，得到該組子圖像數(shù)據(jù)對應的輸出結果數(shù)據(jù)。

這樣，通過將depthwise卷積轉換為普通卷積，能夠利用AI芯片采用對普通卷積的處理方式得到depthwise卷積的處理結果。

在一種可能的實施方式中，所述將所述待處理圖像數(shù)據(jù)分為至少一組子圖像數(shù)據(jù)，包括：基于所述待處理圖像數(shù)據(jù)中的子圖像的數(shù)量、以及預設通道數(shù)量閾值，對所述待處理圖像數(shù)據(jù)中的子圖像進行分組，得到所述至少一組子圖像數(shù)據(jù)。

這樣，基于預設通道數(shù)量閾值對待處理圖像數(shù)據(jù)分組，可以讓每一組子圖像數(shù)據(jù)中包含的子圖像的數(shù)量等于預設通道數(shù)量閾值，進而保證該組子圖像數(shù)據(jù)下的卷積核的每一位置處的卷積操作數(shù)據(jù)都有一個對應的子圖像。

在一種可能的實施方式中，還包括：讀取卷積處理核的配置信息；并基于所述配置信息，確定所述預設通道數(shù)量閾值。

這樣，基于卷積處理核的配置信息得到的預設通道數(shù)量閾值可以知道該卷積處理核中的每一卷積核可以容納卷積操作數(shù)據(jù)的位置的數(shù)量。

在一種可能的實施方式中，針對所述至少一組子圖像數(shù)據(jù)中的每組子圖像數(shù)據(jù)，在該組子圖像數(shù)據(jù)中的子圖像數(shù)量小于所述預設通道數(shù)量閾值的情況下，所述卷積處理方法還包括：為該組子圖像數(shù)據(jù)填充預設圖像數(shù)據(jù)；所述利用該組子圖像數(shù)據(jù)對應的卷積核，對該組子圖像數(shù)據(jù)進行卷積處理，包括：利用該組子圖像數(shù)據(jù)對應的卷積核，對填充了所述預設圖像數(shù)據(jù)后的該組子圖像數(shù)據(jù)進行卷積處理。

這樣，通過在子圖像的數(shù)量小于預設通道數(shù)量閾值的子圖像數(shù)據(jù)中添加預設圖像數(shù)據(jù)，使得該組子圖像數(shù)據(jù)中包含的待處理圖像的子圖像的數(shù)量與預設圖像數(shù)據(jù)中的子圖像的數(shù)量的和等于預設通道數(shù)量閾值。

在一種可能的實施方式中，每組子圖像數(shù)據(jù)對應的卷積核包括：該組子圖像數(shù)據(jù)中的子圖像分別對應的卷積核；所述針對所述至少一組子圖像數(shù)據(jù)中的每組子圖像數(shù)據(jù)，根據(jù)該組子圖像數(shù)據(jù)中各個子圖像分別對應的卷積操作數(shù)，生成該組子圖像數(shù)據(jù)對應的卷積核，包括：針對該組子圖像數(shù)據(jù)中的各子圖像，根據(jù)該子圖像在該組子圖像數(shù)據(jù)中的第一相對位置，確定該子圖像對應的卷積操作數(shù)在對應的卷積核中的第二相對位置；基于所述第二相對位置，生成該子圖像對應的卷積核。