Removed useless snippet that only breaks if batch processing is enabled

pull/3946/head
gilberto.plaza 6 years ago
parent 1c71f00153
commit fa74f691cb
  1. 4
      src/yolo_layer.c

@ -492,7 +492,9 @@ int get_yolo_detections(layer l, int w, int h, int netw, int neth, float thresh,
//printf("\n l.batch = %d, l.w = %d, l.h = %d, l.n = %d \n", l.batch, l.w, l.h, l.n); //printf("\n l.batch = %d, l.w = %d, l.h = %d, l.n = %d \n", l.batch, l.w, l.h, l.n);
int i,j,n; int i,j,n;
float *predictions = l.output; float *predictions = l.output;
if (l.batch == 2) avg_flipped_yolo(l); // This snippet below is not necessary
// Need to comment it in order to batch processing >= 2 images
//if (l.batch == 2) avg_flipped_yolo(l);
int count = 0; int count = 0;
for (i = 0; i < l.w*l.h; ++i){ for (i = 0; i < l.w*l.h; ++i){
int row = i / l.w; int row = i / l.w;

Loading…
Cancel
Save