|
|
@ -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; |
|
|
|