[發明專利]圖像分類方法及裝置、通信設備及存儲介質

在審

• 申請號: 202010980696.4
• 申請日: 2020-09-17
• 公開（公告）號: CN112598016A
• 公開（公告）日: 2021-04-02
• 發明（設計）人: 劉坤
• 申請（專利權）人: 北京小米松果電子有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46
• 代理公司: 北京善任知識產權代理有限公司 11650
• 代理人: 康艷青

1.一種圖像分類方法，其特征在于，包括：

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

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

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

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

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

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

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

3.根據權利要求2所述的方法，其特征在于，所述基于所述第一分類結果，輸出至少指示所述目標圖像為低俗圖像的程度分類信息，包括：

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

和/或，

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

4.根據權利要求1至3任一項所述的方法，其特征在于，所述利用分類模型的特征提取部分提取待分類的目標圖像的語義特征，包括以下至少之一：

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

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

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

5.根據權利要求2所述的方法，其特征在于，在識別所述圖像之前，所述方法還包括：

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

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

在分類模型的訓練損失收斂時，停止訓練；

在所述分類模型的輸出層添加將所述第一分類結果映射為所述第二分類結果的映射層。

6.根據權利要求5所述的方法，其特征在于，所述方法還包括：

對所述第一樣本圖像進行形態變換，得到第二樣本圖像，其中，所述第二樣本圖像與所述第一樣本圖像具有相同的標注標簽；

所述利用所述第一樣本圖像及標注標簽，訓練所述分類模型，包括：

利用所述第一樣本圖像及所述第一樣本圖像的標注標簽，和第二圖像樣本及所述第二樣本圖像的標注標簽，訓練所述分類模型。

7.根據權利要求6所述的方法，其特征在于，所述對所述第一樣本圖像進行形態變換，得到第二樣本圖像，包括以下至少之一：

對所述第一樣本圖像進行線性變換，得到所述第二樣本圖像，其中，所述線性變換包括：圖像旋轉和/或圖像縮放；

對所述第一樣本圖像進行圖像內容變化，得到所述第二樣本圖像，其中，所述圖像內容變化包括：局部的馬賽克處理、局部區域的圖像內容替換和/或局部模糊化處理；

對所述第一樣本圖像進行圖像參數變換，得到所述第二樣本圖像，其中，所述圖像參數包括：對比度和/或亮度。

8.根據權利要求1所述的方法，其特征在于，所述分類模型為SEResNeXt模型。