1. 程式人生 > >mAP的計算方法

mAP的計算方法

Git開源專案(比如unsky的FPN)中test_net.py呼叫test.py下的def test_net(net, imdb, max_per_image=1000, thresh=0.05, vis=False)函式。


程式碼之後會順一遍,先介紹下思路:


假設一張測試圖片有3個標定好的Ground Truth (黑色) 模型預測結果pred為6個框 (5紅 + 1藍)。 首先將所有pred按照置信度(score)反序排列,之後從score最大的pred開始遍歷每個Ground Truth並計算IOU. 如果IOU大於閾值,檢視g+表(最右)對應的GT的det是否為1:不為1則det置1並將此pred的TP至1;det為1則說明之前有更大score的框對應上了這個GT,則此pred的FP置1; 最終根據TP FP計算每個類的AP, 具體過程需要看程式碼

 

Git開源專案(比如unsky的FPN)中test_net.py呼叫test.py下的def test_net(net, imdb, max_per_image=1000, thresh=0.05, vis=False)函式。


程式碼之後會順一遍,先介紹下思路:


假設一張測試圖片有3個標定好的Ground Truth (黑色) 模型預測結果pred為6個框 (5紅 + 1藍)。 首先將所有pred按照置信度(score)反序排列,之後從score最大的pred開始遍歷每個Ground Truth並計算IOU. 如果IOU大於閾值,檢視g+表(最右)對應的GT的det是否為1:不為1則det置1並將此pred的TP至1;det為1則說明之前有更大score的框對應上了這個GT,則此pred的FP置1; 最終根據TP FP計算每個類的AP, 具體過程需要看程式碼