Alexey
2018-01-17 9449156e6d95a672ba3c08f9d7d71bbb75f1a311
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;