[發(fā)明專利]圖像分類方法及裝置、通信設(shè)備及存儲介質(zhì)

在審

• 申請?zhí)枺?/td>: 202010980696.4
• 申請日: 2020-09-17
• 公開（公告）號: CN112598016A
• 公開（公告）日: 2021-04-02
• 發(fā)明（設(shè)計）人: 劉坤
• 申請（專利權(quán)）人: 北京小米松果電子有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46
• 代理公司: 北京善任知識產(chǎn)權(quán)代理有限公司 11650
• 代理人: 康艷青

本公開是關(guān)于一種圖像分類方法及裝置、通信設(shè)備及存儲介質(zhì)。所述圖像分類方法包括：利用分類模型的特征提取部分提取待分類的目標圖像的語義特征；利用所述分類模型的分類部分對所述語義特征進行處理，得到基于M分類的第一分類結(jié)果，其中，所述M為等于或大于3的正整數(shù)；基于所述第一分類結(jié)果，輸出至少指示所述目標圖像為低俗圖像的程度分類信息，其中，所述低俗圖像包括：色情圖像、恐怖圖像和/或暴力圖像；具有實現(xiàn)了目標圖像是否為低俗圖像的判別的同時，實現(xiàn)了圖像的程度細分類，具有分類精確度高的特點，可以滿足細分類場景下的詳細分類要求。

技術(shù)領(lǐng)域

本公開涉及計算機視覺技術(shù)領(lǐng)域，尤其涉及一種圖像分類方法及裝置、通信設(shè)備及存儲介質(zhì)。

背景技術(shù)

隨著用戶產(chǎn)生內(nèi)容(User Generated Content，UGC)內(nèi)容爆發(fā)式增長，信息流產(chǎn)品占領(lǐng)越來越多的用戶時間，而圖片是信息表現(xiàn)最直接的載體。識別和屏蔽低俗內(nèi)容，做好風(fēng)險內(nèi)容控制，凈化圖片載體是一個非常必要的工作。但隨著圖片審核技術(shù)的成熟，越來越多的內(nèi)容創(chuàng)作者在色情擦邊球內(nèi)容創(chuàng)作上愈演愈烈。軟色情和色情，正常和軟色情的邊界越來越模糊，這也給圖片色情審核帶來了很大的困難和挑戰(zhàn)。

發(fā)明內(nèi)容

本申請實施例公開了一種圖像分類方法及裝置、通信設(shè)備及存儲介質(zhì)。

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

利用分類模型的特征提取部分提取待分類的目標圖像的語義特征；

利用所述分類模型的分類部分對所述語義特征進行處理，得到基于M分類的第一分類結(jié)果，其中，所述M為等于或大于3的正整數(shù)；

基于所述第一分類結(jié)果，輸出至少指示所述目標圖像為低俗圖像的程度分類信息，其中，所述低俗圖像包括：色情圖像、恐怖圖像和/或暴力圖像。

基于上述方案，所述方法還包括：

將所述第一分類結(jié)果映射為基于N分類的第二分類結(jié)果，其中，所述N為小于所述M的正整數(shù)；

所述基于N分類的第二分類結(jié)果為：基于父類的分類結(jié)果；

所述基于M分類的第一分類結(jié)果為：基于子類的分類結(jié)果，其中，至少存在一個所述父類包括：多個所述子類。

基于上述方案，所述基于所述第一分類結(jié)果，輸出至少指示所述目標圖像為低俗圖像的程度分類信息，包括：

輸出所述目標圖像的第一分類結(jié)果；

和/或，

輸出所述目標圖像的第二分類結(jié)果。

基于上述方案，所述利用分類模型的特征提取部分提取待分類的目標圖像的語義特征，包括以下至少之一：

針對所述色情圖像，利用所述分類模型的特征提取部分，提取出所述目標圖像中人物元素的衣著特征、行為特征、所述人物元素的特定部位的暴露程度特征及所述人物元素整體暴露程度特征中的一項或多項；

針對所述暴力圖像，利用所述分類模型的特征提取部分，提取出所述目標圖像是否包含有暴力元素、暴力元素的數(shù)量、暴力元素所占圖像面積的比例和暴力元素的暴力程度中的一項或多項；

針對所述恐怖圖像，利用所述分類模型的特征提取部分，提取出所述目標圖像是否包含有恐怖元素、恐怖元素的數(shù)量、恐怖元素所占圖像面積的比例、和恐怖元素的低俗程度中的一項或多項。

基于上述方案，在識別所述圖像之前，所述方法還包括：

獲取第一樣本圖像及標注標簽，其中，所述標注標簽為所述M分類的子類標簽；

利用所述第一樣本圖像及標注標簽，訓(xùn)練所述分類模型；

在分類模型的訓(xùn)練損失收斂時，停止訓(xùn)練；