src/yolo_v2_class.cpp
@@ -154,7 +154,7 @@ cudaSetDevice(net.gpu_index); //std::cout << "net.gpu_index = " << net.gpu_index << std::endl; float nms = .4; //float nms = .4; image im; im.c = img.c; @@ -189,6 +189,7 @@ bbox.h = b.h*im.h; bbox.obj_id = obj_id; bbox.prob = prob; bbox.track_id = 0; bbox_vec.push_back(bbox); }