Loss-graph store automatically (iterations == max_batches) at the end of tr...

AlexeyAB

2018-04-16 c52fa4742866125e35e88d0653dac99473e5fffb

 src/box.c

@@ -246,7 +246,7 @@
    return 0;
}

void do_nms_sort(box *boxes, float **probs, int total, int classes, float thresh)
void do_nms_sort_v2(box *boxes, float **probs, int total, int classes, float thresh)
{
    int i, j, k;
    sortable_bbox *s = calloc(total, sizeof(sortable_bbox));
@@ -292,8 +292,9 @@
   return 0;
}

void do_nms_obj_v3(detection *dets, int total, int classes, float thresh)
void do_nms_obj(detection *dets, int total, int classes, float thresh)
{
   printf(" total = %d, classes = %d, thresh = %f \n", total, classes, thresh);
   int i, j, k;
   k = total - 1;
   for (i = 0; i <= k; ++i) {
@@ -328,7 +329,7 @@
   }
}

void do_nms_sort_v3(detection *dets, int total, int classes, float thresh)
void do_nms_sort(detection *dets, int total, int classes, float thresh)
{
   int i, j, k;
   k = total - 1;