little fix

This commit is contained in:
Joseph Redmon 2017-04-09 20:15:22 -07:00
parent e610c7de9a
commit 77ee1118bc

View File

@ -347,6 +347,9 @@ void get_region_boxes(layer l, int w, int h, float thresh, float **probs, box *b
int col = i % l.w;
for(n = 0; n < l.n; ++n){
int index = n*l.w*l.h + i;
for(j = 0; j < l.classes; ++j){
probs[index][j] = 0;
}
int obj_index = entry_index(l, 0, n*l.w*l.h + i, 4);
int box_index = entry_index(l, 0, n*l.w*l.h + i, 0);
float scale = predictions[obj_index];