diff --git a/src/yolo_layer.c b/src/yolo_layer.c index a76b5efb..20ee8e34 100644 --- a/src/yolo_layer.c +++ b/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); int i,j,n; 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; for (i = 0; i < l.w*l.h; ++i){ int row = i / l.w;