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

Fixed a bug in the object_detector testing code that caused the average 

precision values to be slightly wrong.
This commit is contained in:
Davis King 2013-07-31 19:24:42 -04:00
parent c0c1995d50
commit 92a3c41010
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
dlib/svm/cross_validate_object_detection_trainer.h
View File

@ -50,32 +50,27 @@ namespace dlib
std::vector<bool> used(boxes.size(),false);
for (unsigned long i = 0; i < truth_boxes.size(); ++i)
{
unsigned long best_idx = 0;
double best_overlap = 0;
// Find the best box that hits truth_boxes[i]
bool found_match = false;
// Find the first box that hits truth_boxes[i]
for (unsigned long j = 0; j < boxes.size(); ++j)
{
if (used[j])
continue;
const double overlap = truth_boxes[i].get_rect().intersect(boxes[j].second).area() / (double)(truth_boxes[i].get_rect()+boxes[j].second).area();
if (overlap > best_overlap)
const double overlap = truth_boxes[i].get_rect().intersect(boxes[j].second).area() /
(double)(truth_boxes[i].get_rect()+boxes[j].second).area();
if (overlap >= overlap_eps)
{
best_overlap = overlap;
best_idx = j;
used[j] = true;
++count;
found_match = true;
break;
}
}
// if there was any box that hit truth_boxes[i]
if (best_overlap > overlap_eps)
{
used[best_idx] = true;
++count;
}
else
{
if (!found_match)
++missing_detections;
}
}
for (unsigned long i = 0; i < boxes.size(); ++i)