AlexeyAB
2018-03-01 35a66b73a25ab93f2a3ceb85993d4801240046a4
src/detector.c
@@ -245,8 +245,8 @@
        if (ymax > h) ymax = h;
        for(j = 0; j < classes; ++j){
            int class = j;
            if (probs[i][class]) fprintf(fp, "%d %d %f %f %f %f %f\n", id, j+1, probs[i][class],
            int class_id = j;
            if (probs[i][class_id]) fprintf(fp, "%d %d %f %f %f %f %f\n", id, j+1, probs[i][class_id],
                    xmin, ymin, xmax, ymax);
        }
    }
@@ -777,7 +777,7 @@
         avg_precision += cur_precision;
      }
      avg_precision = avg_precision / 11;
      printf("class = %d, name = %s, \t ap = %2.2f %% \n", i, names[i], avg_precision*100);
      printf("class_id = %d, name = %s, \t ap = %2.2f %% \n", i, names[i], avg_precision*100);
      mean_average_precision += avg_precision;
   }
   
@@ -859,6 +859,7 @@
void run_detector(int argc, char **argv)
{
   int http_stream_port = find_int_arg(argc, argv, "-http_port", -1);
   char *out_filename = find_char_arg(argc, argv, "-out_filename", 0);
    char *prefix = find_char_arg(argc, argv, "-prefix", 0);
    float thresh = find_float_arg(argc, argv, "-thresh", .24);
@@ -911,6 +912,6 @@
        char **names = get_labels(name_list);
      if(filename)
         if (filename[strlen(filename) - 1] == 0x0d) filename[strlen(filename) - 1] = 0;
        demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename);
        demo(cfg, weights, thresh, cam_index, filename, names, classes, frame_skip, prefix, out_filename, http_stream_port);
    }
}