OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #6151 from adityap27/master 

Fixed incorrect calculation of mAP for -points 0 (for own dataset) issue #6040
This commit is contained in:
Alexey 2020-07-06 21:36:48 +03:00 committed by GitHub
parent 00919eeb99 7753a965b4
commit d8254f2212
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/detector.c
View File

@ -1253,6 +1253,12 @@ float validate_detector_map(char *datacfg, char *cfgfile, char *weightfile, floa
avg_precision += delta_recall * last_precision; avg_precision += delta_recall * last_precision;
} }
//add remaining area of PR curve when recall isn't 0 at rank-1
if(pr[i][0].recall!=0)
{
//no need of delta_recall=last_recall-0;
avg_precision += last_recall * last_precision;
}
} }
// MSCOCO - 101 Recall-points, PascalVOC - 11 Recall-points // MSCOCO - 101 Recall-points, PascalVOC - 11 Recall-points
else else