Fixed detector recall

pull/604/merge
AlexeyAB 7 years ago
parent c3c6703b8b
commit 726cebd3fb
  1. 5
      build/darknet/x64/darknet_web_cam_voc.cmd
  2. 4
      src/detector.c

@ -1,6 +1,9 @@
# play video from Web-Camera number 0
darknet.exe detector demo data/voc.data cfg/yolov2-voc.cfg yolo-voc.weights -c 0
rem darknet.exe detector demo data/voc.data cfg/yolov2-voc.cfg yolo-voc.weights -c 0
darknet.exe detector demo data/coco.data cfg/yolov3.cfg yolov3.weights -c 0
pause

@ -471,7 +471,7 @@ void validate_detector_recall(char *datacfg, char *cfgfile, char *weightfile)
++total;
box t = { truth[j].x, truth[j].y, truth[j].w, truth[j].h };
float best_iou = 0;
for (k = 0; k < l.w*l.h*l.n; ++k) {
for (k = 0; k < nboxes; ++k) {
float iou = box_iou(dets[k].bbox, t);
if (dets[k].objectness > thresh && iou > best_iou) {
best_iou = iou;
@ -828,7 +828,7 @@ int anchors_comparator(const void *pa, const void *pb)
{
anchors_t a = *(anchors_t *)pa;
anchors_t b = *(anchors_t *)pb;
float diff = b.w - a.w;
float diff = b.w*b.h - a.w*a.h;
if (diff < 0) return 1;
else if (diff > 0) return -1;
return 0;

Loading…
Cancel
Save