| src/yolo_v2_class.cpp | ●●●●● patch | view | raw | blame | history |
src/yolo_v2_class.cpp
@@ -180,11 +180,12 @@ cudaGetDevice(&old_gpu_index); if(cur_gpu_id != old_gpu_index) cudaSetDevice(net.gpu_index); net.wait_stream = wait_stream; // 1 - wait CUDA-stream, 0 - not to wait #endif //std::cout << "net.gpu_index = " << net.gpu_index << std::endl; //float nms = .4; net.wait_stream = wait_stream; // 1 - wait CUDA-stream, 0 - not to wait image im; im.c = img.c;