AlexeyAB
2018-01-17 5a2a5679708c098a9d9d5973bc485f2e3c35c61d
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;