[發明專利]圖像分類方法、圖像分類網絡訓練方法、裝置及電子設備

在審

• 申請號: 202111489645.2
• 申請日: 2021-12-08
• 公開（公告）號: CN114120044A
• 公開（公告）日: 2022-03-01
• 發明（設計）人: 陳圣;曾定衡;王洪斌;周迅溢
• 申請（專利權）人: 馬上消費金融股份有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/774;G06V10/80;G06V10/772;G06V10/82;G06N3/04;G06N3/08;G06K9/62
• 代理公司: 北京銀龍知識產權代理有限公司 11243
• 代理人: 胡永芳

本申請提供一種圖像分類方法、圖像分類網絡訓練方法、裝置及電子設備，圖像分類方法，包括：獲取待分類圖像；將待分類圖像輸入目標圖像分類網絡中進行分類，得到目標分類結果；其中，目標圖像分類網絡包括第一卷積層、加權融合模塊、第一相加層、第二卷積層以及分類層，第一卷積層的輸入包括待分類圖像，第一卷積層的輸出還與第一相加層的輸入連接。即采用一個目標圖像分類網絡進行分類即可，目標圖像分類網絡的結構復雜度較低，可減少分類花費的時間，從而可提高圖像分類效率。

技術領域

本申請涉及圖像處理技術領域，尤其涉及到一種圖像分類方法、圖像分類網絡訓練方法、裝置及電子設備。

背景技術

圖像分類是圖像處理中常見的一種技術，目前常見的圖像分類方法包括基于特征提取的傳統分類方法以及基于深度學習的分類方法等；

其中，基于深度學習的方法中，常用的方法是采用兩個卷積神經網絡分別進行處理，然后對兩個卷積神經網絡得到的特征進行匯聚，并進行分類，這樣，容易導致處理效率較差。

發明內容

本申請實施例提供一種圖像分類方法、圖像分類網絡訓練方法、裝置及電子設備，以解決處理效率較差的問題。

為了解決上述技術問題，本申請實施例是這樣實現的：

第一方面，本申請實施例提供了一種圖像分類方法，包括：

獲取待分類圖像；

將所述待分類圖像輸入目標圖像分類網絡中進行分類，得到目標分類結果；

其中，所述目標圖像分類網絡包括第一卷積層、加權融合模塊、第一相加層、第二卷積層以及分類層，所述第一卷積層的輸入包括所述待分類圖像，所述第一卷積層的輸出還與所述第一相加層的輸入連接。

第二方面，本申請實施例提供了一種圖像分類網絡訓練方法，包括：

獲取訓練樣本圖像集；

將所述訓練樣本圖像集輸入初始分類網絡進行訓練，得到目標圖像分類網絡；

其中，所述初始分類網絡包括第一初始卷積層、初始加權融合模塊、第一初始相加層、第二初始卷積層以及初始分類層，所述第一初始卷積層的輸入包括所述訓練樣本圖像，所述第一初始卷積層的輸出還與所述第一初始相加層的輸入連接。

第三方面，本申請實施例還提供一種圖像分類裝置，包括：

第一獲取模塊，用于獲取待分類圖像集；

分類模塊，用于將所述待分類圖像集輸入目標圖像分類網絡中進行分類，得到目標分類結果；

其中，所述目標圖像分類網絡包括第一卷積層、加權融合模塊、第一相加層、第二卷積層以及分類層，所述第一卷積層的輸入包括所述待分類圖像，所述第一卷積層的輸出還與所述第一相加層的輸入連接。

第四方面，本申請實施例還提供一種圖像分類網絡訓練裝置，包括：

第二獲取模塊，用于獲取訓練樣本圖像；

訓練模塊，用于將所述訓練樣本圖像輸入初始分類網絡進行訓練，得到目標圖像分類網絡；

其中，所述初始分類網絡包括依次連接的第一卷積層、加權融合模塊、第一相加層、第二卷積層以及分類層，所述第一卷積層的輸入包括所述訓練樣本圖像，所述第一卷積層的輸出還與所述第一相加層的輸入連接。

第五方面，本申請實施例還提供一種電子設備，包括：存儲器、處理器及存儲在所述存儲器上并可在所述處理器上運行的計算機程序，所述處理器執行所述計算機程序時實現上述圖像分類網絡訓練方法或者圖像分類方法中的步驟。