[發明專利]痤瘡檢測方法、痤瘡檢測模型訓練方法及相關裝置

在審

• 申請號: 202011012179.4
• 申請日: 2020-09-22
• 公開（公告）號: CN112364685A
• 公開（公告）日: 2021-02-12
• 發明（設計）人: 陳仿雄
• 申請（專利權）人: 深圳數聯天下智能科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/32;G06K9/62;G06N3/04;G06N3/08;G06N20/20
• 代理公司: 深圳中細軟知識產權代理有限公司 44528
• 代理人: 袁文英

1.一種痤瘡檢測方法，其特征在于，所述方法包括：

獲取待檢測痤瘡的人臉圖像；

將所述人臉圖像輸入至痤瘡檢測模型，獲取所述人臉圖像的痤瘡檢測集合，所述痤瘡檢測集合包含檢測到的邊界框，及所述邊界框的檢測值，所述檢測值用于指示所述邊界框內檢測到痤瘡的可能性；

確定所述痤瘡檢測集合中的重疊框集合，并對所述重疊框集合中的邊界框的檢測值進行更新，根據更新后的重疊框集合確定目標邊界框，并從所述痤瘡檢測集合中提取所述目標邊界框，將所述目標邊界框添加至目標邊界框集合，所述重疊框集合包含標識同一個痤瘡區域的至少兩個邊界框，所述目標邊界框為能夠代表所述痤瘡區域的最優的邊界框；

根據所述目標邊界框集合確定所述人臉圖像的痤瘡檢測結果。

2.根據權利要求1所述的方法，其特征在于，所述確定所述痤瘡檢測集合中的重疊框集合，并對所述重疊框集合中的邊界框的檢測值進行更新，根據更新后的重疊框集合確定目標邊界框，并從所述痤瘡檢測集合中提取所述目標邊界框，將所述目標邊界框添加至目標邊界框集合，包括：

計算所述痤瘡檢測集合中第一邊界框與其他邊界框的重疊比例，基于重疊比例確定所述重疊框集合；所述第一邊界框為所述痤瘡檢測集合中檢測值最大的邊界框，所述其他邊界框為所述痤瘡檢測集合中除所述第一邊界框以外的其他任一邊界框，所述重疊框集合中包含所述第一邊界框，及與所述第一邊界框的重疊比例大于或等于第一預設比例值的目標其他邊界框；

根據所述目標其他邊界框與所述第一邊界框的重疊比例，對所述重疊框集合中的目標其他邊界框的檢測值進行更新，得到更新后的重疊框集合；

從所述痤瘡檢測集合中提取出所述更新后的重疊框集合中具有最大檢測值的邊界框作為目標邊界框，并添加至所述目標邊界框集合中；

返回執行所述計算所述痤瘡檢測集合中第一邊界框與其他邊界框的重疊比例，基于重疊比例確定所述重疊框集合的步驟，直至所述重疊框集合中所述目標其他邊界框的數量為零。

3.根據權利要求2所述的方法，其特征在于，所述根據所述目標其他邊界框與所述第一邊界框的重疊比例，對所述重疊框集合中的目標其他邊界框的檢測值進行更新，包括：

當所述目標其他邊界框與所述第一邊界框的重疊比例大于或等于所述第一預設比例值，且小于第二預設比例值時，將所述目標其他邊界框的預測值的對數值與預設標準值的和作為所述目標其他邊界框更新后的預測值；

當所述目標其他邊界框與所述第一邊界框的重疊比例大于或等于所述第二預設比例值時，計算所述目標其他邊界框與所述第一邊界框的重疊比例與所述預設標準值之間的差值的絕對值，將所述絕對值與所述目標其他邊界框的預測值的乘積作為所述目標其他邊界框更新后的預測值。

4.根據權利要求1所述的方法，其特征在于，所述痤瘡檢測模型包括yolov3模型及三個空間金字塔池化層，三個所述空間金字塔池化層分別連接至所述yolov3模型的三個輸出分支的末端。

5.根據權利要求1所述的方法，其特征在于，所述將所述人臉圖像輸入至痤瘡檢測模型，獲取所述人臉圖像的痤瘡檢測集合，包括：

將所述人臉圖像輸入所述痤瘡檢測模型，獲得所述痤瘡檢測模型輸出的目標特征圖；

根據所述目標特征圖及預設的錨點框，得到所述人臉圖像的痤瘡檢測集合，所述錨點框是根據預設的倍數值、基于痤瘡樣本數據集中實際邊界框確定的所述實際邊界框長度的第一最大值和第一最小值，及寬度的第二最大值和第二最小值得到的，所述痤瘡樣本數據集為用于訓練得到所述痤瘡檢測模型的數據，所述痤瘡樣本數據集包含痤瘡樣本圖像，及所述痤瘡樣本圖像已標注的表示痤瘡所在位置的所述實際邊界框。