Philip Kahn
2018-05-04 3459c47bfa02d23d300df9a44ba3ec9c958850fe
src/detector.c
@@ -1,3 +1,8 @@
#ifdef _DEBUG
#include <stdlib.h>
#include <crtdbg.h>
#endif
#include "network.h"
#include "region_layer.h"
#include "cost_layer.h"
@@ -86,6 +91,7 @@
    args.n = imgs;
    args.m = plist->size;
    args.classes = classes;
    args.flip = net.flip;
    args.jitter = jitter;
    args.num_boxes = l.max_boxes;
   args.small_object = net.small_object;
@@ -644,6 +650,8 @@
            truth_dif = read_boxes(labelpath_dif, &num_labels_dif);
         }
         const int checkpoint_detections_count = detections_count;
         for (i = 0; i < nboxes; ++i) {
            int class_id;
@@ -694,7 +702,13 @@
                  // calc avg IoU, true-positives, false-positives for required Threshold
                  if (prob > thresh_calc_avg_iou) {
                     if (truth_index > -1) {
                     int z, found = 0;
                     for (z = checkpoint_detections_count; z < detections_count-1; ++z)
                        if (detections[z].unique_truth_index == truth_index) {
                           found = 1; break;
                        }
                     if(truth_index > -1 && found == 0) {
                        avg_iou += max_iou;
                        ++tp_for_thresh;
                     }
@@ -714,7 +728,8 @@
      }
   }
   avg_iou = avg_iou / (tp_for_thresh + fp_for_thresh);
   if((tp_for_thresh + fp_for_thresh) > 0)
      avg_iou = avg_iou / (tp_for_thresh + fp_for_thresh);
   
   // SORT(detections)
@@ -1059,8 +1074,8 @@
        }
        image im = load_image_color(input,0,0);
      int letterbox = 0;
        image sized = resize_image(im, net.w, net.h);
      //image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        //image sized = resize_image(im, net.w, net.h);
      image sized = letterbox_image(im, net.w, net.h); letterbox = 1;
        layer l = net.layers[net.n-1];
        //box *boxes = calloc(l.w*l.h*l.n, sizeof(box));
@@ -1069,8 +1084,8 @@
        float *X = sized.data;
        time= what_time_is_it_now();
        //network_predict(net, X);
      network_predict_image(&net, im); letterbox = 1;
        network_predict(net, X);
      //network_predict_image(&net, im); letterbox = 1;
        printf("%s: Predicted in %f seconds.\n", input, (what_time_is_it_now()-time));
        //get_region_boxes(l, 1, 1, thresh, probs, boxes, 0, 0);
      // if (nms) do_nms_sort_v2(boxes, probs, l.w*l.h*l.n, l.classes, nms);
@@ -1097,6 +1112,22 @@
#endif
        if (filename) break;
    }
   // free memory
   free_ptrs(names, net.layers[net.n - 1].classes);
   free_list(options);
   int i;
   const int nsize = 8;
   for (j = 0; j < nsize; ++j) {
      for (i = 32; i < 127; ++i) {
         free_image(alphabet[j][i]);
      }
      free(alphabet[j]);
   }
   free(alphabet);
   free_network(net);
}
void run_detector(int argc, char **argv)