Temporary fix

pull/604/merge
AlexeyAB 7 years ago
parent 6d56c38e8b
commit d0039f6dfd
  1. 2
      src/network.c

@ -543,7 +543,7 @@ void custom_get_region_detections(layer l, int w, int h, int net_w, int net_h, f
float **probs = calloc(l.w*l.h*l.n, sizeof(float *));
int i, j;
for (j = 0; j < l.w*l.h*l.n; ++j) probs[j] = calloc(l.classes, sizeof(float *));
get_region_boxes(l, w, h, thresh, probs, boxes, 0, map);
get_region_boxes(l, 1, 1, thresh, probs, boxes, 0, map);
for (j = 0; j < l.w*l.h*l.n; ++j) {
dets[j].classes = l.classes;
dets[j].bbox = boxes[j];

Loading…
Cancel
Save