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;
