AlexeyAB
2018-01-17 5a2a5679708c098a9d9d5973bc485f2e3c35c61d
Fixed wait_stream flag, for compilation without GPU
1 files modified
3 ■■■■ changed files
src/yolo_v2_class.cpp 3 ●●●● 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;