[發明專利]一種色情圖像識別方法、系統、存儲介質和電子設備

在審

• 申請號: 202110863334.1
• 申請日: 2021-07-29
• 公開（公告）號: CN113537380A
• 公開（公告）日: 2021-10-22
• 發明（設計）人: 余路陽;唐會軍;劉拴林;梁堃;陳建
• 申請（專利權）人: 北京數美時代科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/46;G06N3/04;G06N3/08
• 代理公司: 北京輕創知識產權代理有限公司 11212
• 代理人: 翟磊

1.一種色情圖像識別方法，其特征在于，包括：

將待識別圖像輸入雙分支訓練的色情圖像識別模型，得到每個分支對應的結果，其中，第一分支對應的結果包括至少一個第一色情標簽，第二分支的結果只包括一個第二色情標簽；

根據第一分支對應的結果和第二分支對應的結果判斷所述待識別圖像是否為色情圖像。

2.根據權利要求1所述的一種色情圖像識別方法，其特征在于，還包括：

對任一預設圖像標注至少一個第一色情標簽，以及對所述任一預設圖像只標注一個第二色情標簽，直至得到每個預設圖像對應的至少一個第一色情標簽，以及，每個預設圖像對應的一個第二色情標簽；

搭建包括特征提取器和分類器的改進inception v3模型，其中，所述分類器包括所述第一分支和所述第二分支，其中，所述特征提取器將根據每個預設圖像、每個預設圖像對應的至少一個第一色情標簽，以及每個預設圖像對應的一個第二色情標簽得到的特征向量分別輸入所述第一分支和所述第二分支；

基于每個預設圖像對應的至少一個第一色情標簽，以及，每個預設圖像對應的第二色情標簽，利用所述改進inception v3模型，并結合Adam算法進行訓練，得到所述色情圖像識別模型。

3.根據權利要求1或2所述的一種色情圖像識別方法，其特征在于，搭建所述改進inception v3模型，還包括：

對所述第一分支引入binary-cross-entropy損失函數，對所述第二分支引入cross-entropy損失函數。

4.根據權利要求1或2所述的一種色情圖像識別方法，其特征在于，所述第一分支包括兩個不同維數的全連接層，且所述第一分支對應的激活函數為sigmoid函數；

所述第二分支包括兩個不同維數的全連接層，且所述第二分支對應的激活函數為softmax函數。

5.一種色情圖像識別系統，其特征在于，包括輸入模塊和判斷模塊；

所述輸入模塊用于將待識別圖像輸入雙分支訓練的色情圖像識別模型，得到每個分支對應的結果，其中，第一分支對應的結果包括至少一個第一色情標簽，第二分支的結果只包括一個第二色情標簽；

所述判斷模塊用于根據第一分支對應的結果和第二分支對應的結果判斷所述待識別圖像是否為色情圖像。

6.根據權利要求5所述的一種色情圖像識別系統，其特征在于，還包括標注模塊、模型搭建模塊和模型訓練模塊；

所述標注模塊用于對任一預設圖像標注至少一個第一色情標簽，以及對所述任一預設圖像只標注一個第二色情標簽，直至得到每個預設圖像對應的至少一個第一色情標簽，以及，每個預設圖像對應的一個第二色情標簽；

所述模型搭建模塊用于搭建包括特征提取器和分類器的改進inception v3模型，其中，所述分類器包括所述第一分支和所述第二分支，其中，所述特征提取器將根據每個預設圖像、每個預設圖像對應的至少一個第一色情標簽，以及每個預設圖像對應的一個第二色情標簽得到的特征向量分別輸入所述第一分支和所述第二分支；

所述模型訓練模塊用于基于每個預設圖像對應的至少一個第一色情標簽，以及，每個預設圖像對應的第二色情標簽，利用所述改進inception v3模型，并結合Adam算法進行訓練，得到所述色情圖像識別模型。

7.根據權利要求5或6所述的一種色情圖像識別系統，其特征在于，所述模型搭建模塊還用于：

對所述第一分支引入binary-cross-entropy損失函數，對所述第二分支引入cross-entropy損失函數。

8.根據權利要求5或6所述的一種色情圖像識別系統，其特征在于，所述第一分支包括兩個不同維數的全連接層，且所述第一分支對應的激活函數為sigmoid函數；

所述第二分支包括兩個不同維數的全連接層，且所述第二分支對應的激活函數為softmax函數。