AlexeyAB
2018-06-04 efaf684cb20b996db0cd6d99d20fabb4bc26859a
src/yolo_v2_class.cpp
@@ -69,6 +69,7 @@
   }
   set_batch_network(&net, 1);
   net.gpu_index = cur_gpu_id;
   fuse_conv_batchnorm(net);
   layer l = net.layers[net.n - 1];
   int j;
@@ -221,7 +222,7 @@
   int letterbox = 0;
   float hier_thresh = 0.5;
   detection *dets = get_network_boxes(&net, im.w, im.h, thresh, hier_thresh, 0, 1, &nboxes, letterbox);
   if (nms) do_nms_sort_v3(dets, nboxes, l.classes, nms);
   if (nms) do_nms_sort(dets, nboxes, l.classes, nms);
   std::vector<bbox_t> bbox_vec;