OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #3946 from keko950/batch 

Removed useless snippet that only breaks if batch processing is enabled
This commit is contained in:
Alexey 2019-09-18 15:36:58 +03:00 committed by GitHub
parent 1c71f00153 fa74f691cb
commit b918bf0329
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/yolo_layer.c
View File

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