[發明專利]用于色情圖片的圖像數據處理方法及裝置

在審

• 申請號: 201811221092.0
• 申請日: 2018-10-19
• 公開（公告）號: CN109508643A
• 公開（公告）日: 2019-03-22
• 發明（設計）人: 郭英強;張默
• 申請（專利權）人: 北京陌上花科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 北京卓唐知識產權代理有限公司 11541
• 代理人: 唐海力;李志剛

本申請公開了一種用于色情圖片的圖像數據處理方法及裝置。該方法包括生成訓練數據集，其中，訓練數據集包括：預設人體關鍵點標簽；根據所述訓練數據集訓練得到預訓練模型；輸入待檢測圖片；以及根據預訓練模型判斷所述待檢測圖片是否屬于色情圖片。本申請解決了對于色情圖片的檢測處理效果較差的技術問題。通過本申請的方法采用改進的生成對抗網絡可批量生成一批包含有正常、色情/性感兩種類別的圖片集。可大大減少人力成本，并且非常高效。得到的圖片集可快速區分待測圖片是否屬于色情/性感圖片。

技術領域

本申請涉及計算機視覺領域，具體而言，涉及一種用于色情圖片的圖像數據處理方法及裝置。

背景技術

網絡圖片是海量豐富的，但圖片的質量參差不齊，并且沒有標簽。

發明人發現，針對包含有性感、色情圖片中人體姿態多變，不易被辨別。如果想要訓練模型需要人為標注的圖片費時、費力，而且成本較高、效率低下。

針對相關技術中對于色情圖片的檢測處理效果較差的問題，目前尚未提出有效的解決方案。

發明內容

本申請的主要目的在于提供一種用于色情圖片的圖像數據處理方法及裝置，以解決對于色情圖片的檢測處理效果較差的問題。

為了實現上述目的，根據本申請的一個方面，提供了一種用于色情圖片的圖像數據處理方法。

根據本申請的用于色情圖片的圖像數據處理方法包括：生成訓練數據集，其中，訓練數據集包括：預設人體關鍵點標簽；根據所述訓練數據集訓練得到預訓練模型；輸入待檢測圖片；以及根據預訓練模型判斷所述待檢測圖片是否屬于色情圖片。

進一步地，成訓練數據集包括：提取包含有相關要素的圖片中的人體關鍵點信息；根據所述人體關鍵點信息對人體執行裁剪操作；以及將裁剪得到的圖片和對應的已標注有相關要素的標簽儲存到訓練集數據庫。

進一步地，生成訓練數據集包括：根據關鍵點對人體執行裁剪；以及將人體圖像中的肩部位置的關鍵點在原圖中的中央位置對齊。

進一步地，根據預訓練模型判斷所述待檢測圖片是否屬于色情圖片包括：在預訓練模型中輸入與訓練階段相同分布的隨機變量及擬得到的圖片標簽；以及根據預訓練模型中生成的對抗網絡，獲取生成的圖片。

進一步地，根據預訓練模型判斷所述待檢測圖片是否屬于色情圖片還包括：模型訓練過程，所述模型訓練過程包括：在生成網絡中生成類似真實的預設目標圖片；在判斷網絡中判斷一個給定預設圖片是不是真實的預設目標圖片；以及將上述生成網絡和判斷網絡持續對抗訓練，并通過預測的判決結果與真實的結果之間的差別，更新生成網絡和判決網絡中的參數。

為了實現上述目的，根據本申請的另一方面，提供了一種用于色情圖片的圖像數據處理裝置。

根據本申請的用于色情圖片的圖像數據處理裝置包括：生成模塊，用于生成訓練數據集，其中，訓練數據集包括：預設人體關鍵點標簽；訓練模塊，用于根據所述訓練數據集訓練得到預訓練模型；輸入模塊，用于輸入待檢測圖片；以及判斷模塊，用于根據預訓練模型判斷所述待檢測圖片是否屬于色情圖片。

進一步地，所述生成模塊包括：提取單元，用于提取包含有相關要素的圖片中的人體關鍵點信息；裁剪單元，用于根據所述人體關鍵點信息對人體執行裁剪操作；以及存儲單元，用于將裁剪得到的圖片和對應的已標注有相關要素的標簽儲存到訓練集數據庫。

進一步地，所述生成模塊包括：關鍵點裁剪單元，用于根據關鍵點對人體執行裁剪；以及對齊單元，用于將人體圖像中的肩部位置的關鍵點在原圖中的中央位置對齊。

進一步地，所述判斷模塊包括：變量和標簽輸入單元，用于在預訓練模型中輸入與訓練階段相同分布的隨機變量及擬得到的圖片標簽；以及判決單元，用于根據預訓練模型中生成的對抗網絡，獲取生成的圖片。