[發(fā)明專利]一種應(yīng)用于人臉檢測系統(tǒng)的基于蒙版的模型評估方法

有效

• 申請?zhí)枺?/td>: 202011091940.8
• 申請日: 2020-10-13
• 公開（公告）號: CN112215154B
• 公開（公告）日: 2021-05-25
• 發(fā)明（設(shè)計）人: 孫家樂;瞿洪桂;袁麗燕;朱海明;高云麗
• 申請（專利權(quán)）人: 北京中電興發(fā)科技有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62
• 代理公司: 北京市盛峰律師事務(wù)所 11337
• 代理人: 于國強

1.一種應(yīng)用于人臉檢測系統(tǒng)的基于蒙版的模型評估方法，其特征在于，包括以下步驟：

S1)獲取測試圖片以及待評估人臉檢測模型，獲取所述測試圖片中的所有標(biāo)注框，根據(jù)環(huán)境因素將測試圖片中的所有標(biāo)注框劃分為若干個標(biāo)注人臉框和若干個標(biāo)注蒙版框；

S2)利用待評估人臉檢測模型對測試圖片進行人臉檢測，得到測試圖片中的所有預(yù)測框以及每個預(yù)測框的預(yù)測框信息；

S3)計算測試圖片中所有預(yù)測框與所述測試圖片中的所有標(biāo)注框之間的IoU值，設(shè)置n個不同的IoU閾值；

S4)在第s個IoU閾值下設(shè)置置信度閾值取值范圍,在所述置信度閾值取值范圍內(nèi)進行等間距取值，獲得m個置信度閾值，s＝1、2、...、n；

S5)篩選出測試圖片中置信度大于第i個置信度閾值的若干個預(yù)測框，i＝1、2、...、m；

S6)根據(jù)IoU值將測試圖片中置信度大于第i個置信度閾值的若干個預(yù)測框分別與步驟S1)中的若干個標(biāo)注人臉框和若干個標(biāo)注蒙版框進行匹配，獲得匹配結(jié)果，包括以下步驟：

S61)按照置信度從高到低的順序?qū)y試圖片中置信度大于第i個置信度閾值的若干個預(yù)測框進行排序，獲得排序后的若干個預(yù)測框；

S62)計算排序后的若干個預(yù)測框中的第j個預(yù)測框分別與測試圖片中所有未曾匹配成功過的標(biāo)注人臉框之間的IoU值，獲得第一IoU最大值下標(biāo)z表示測試圖片中所有未曾匹配成功過的標(biāo)注人臉框中的第z個標(biāo)注人臉框，判斷所述最大值是否大于所述第s個IoU閾值，若是，則標(biāo)注排序后的若干個預(yù)測框中的第j個預(yù)測框與測試圖片中所有未曾匹配成功過的標(biāo)注人臉框中的第z個標(biāo)注人臉框匹配成功，進入步驟S64)；若否，則標(biāo)注排序后的若干個預(yù)測框中的第j個預(yù)測框匹配失敗，進入步驟S63)；

S63)計算排序后的若干個預(yù)測框中的第j個預(yù)測框分別與測試圖片中所有未曾匹配成功過的標(biāo)注蒙版框之間的IoU值，獲得第二最大值下標(biāo)f表示測試圖片中所有未曾匹配成功過的標(biāo)注蒙版框的第f個標(biāo)注蒙版框，判斷所述第二最大值是否大于所述第s個IoU閾值，若是，則標(biāo)注測試圖片中置信度大于所述第i個置信度閾值的第j個預(yù)測框與測試圖片中所有未曾匹配成功過的標(biāo)注蒙版框的第f個標(biāo)注蒙版框匹配成功；若否，則標(biāo)注測試圖片中置信度大于所述第i個置信度閾值的第j個預(yù)測框匹配失敗，進入步驟S64)；

S64)依次重復(fù)步驟S62)至S63)，按照置信度從高到低的順序遍歷排序后的若干個預(yù)測框，獲得排序后的若干個預(yù)測框的匹配結(jié)果；

根據(jù)所述匹配結(jié)果建立混淆矩陣，根據(jù)所述混淆矩陣計算與第i個置信度閾值相對應(yīng)的精確度和召回率；

S7)依次重復(fù)步驟S5)至步驟S6)，在第s個IoU閾值下遍歷所有置信度閾值，獲得在第s個IoU閾值下與所有置信度閾值分別相對應(yīng)的精確度和召回率，根據(jù)所述在第s個IoU閾值下與所有置信度閾值分別相對應(yīng)的精確度和召回率繪制P-R曲線，根據(jù)P-R曲線獲得與第s個IoU閾值相對應(yīng)的AP值；

S8)依次重復(fù)步驟S4)至步驟S7)，遍歷所有不同的IoU閾值，獲得與所有不同的IoU閾值分別對應(yīng)的若干個AP值；

S9)計算步驟S10)中若干個AP值的平均值，將所述若干個AP值的平均值作為人臉檢測模型的評估指標(biāo)結(jié)果。

2.根據(jù)權(quán)利要求1所述的應(yīng)用于人臉檢測系統(tǒng)的基于蒙版的模型評估方法，其特征在于，在步驟S6)中，根據(jù)所述匹配結(jié)果建立混淆矩陣，根據(jù)所述混淆矩陣計算與第i個置信度閾值相對應(yīng)的精確度和召回率，包括以下步驟：

S65)統(tǒng)計標(biāo)注人臉框的正檢數(shù)TP_i、標(biāo)注人臉框的漏檢數(shù)FN_i和標(biāo)注人臉框的誤檢數(shù)FP_i；

S66)根據(jù)所述標(biāo)注人臉框的正檢數(shù)TP_i、所述標(biāo)注人臉框的漏檢數(shù)FN_i和所述標(biāo)注人臉框的誤檢數(shù)FP_i建立混淆矩陣；

S67)獲得與第i個置信度閾值相對應(yīng)的精確度獲得與第i個置信度閾值相對應(yīng)的召回率