mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

Fixed wait_stream flag, for compilation without GPU

Browse Source
This commit is contained in:
AlexeyAB
2018-01-17 12:56:51 +03:00
parent 270a38ecad
commit 5a2a567970
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/yolo_v2_class.cpp
View File

@ -180,11 +180,12 @@ YOLODLL_API std::vector<bbox_t> Detector::detect(image_t img, float thresh, bool
cudaGetDevice(&old_gpu_index); cudaGetDevice(&old_gpu_index);
if(cur_gpu_id != old_gpu_index) if(cur_gpu_id != old_gpu_index)
cudaSetDevice(net.gpu_index); cudaSetDevice(net.gpu_index);
net.wait_stream = wait_stream; // 1 - wait CUDA-stream, 0 - not to wait
#endif #endif
//std::cout << "net.gpu_index = " << net.gpu_index << std::endl; //std::cout << "net.gpu_index = " << net.gpu_index << std::endl;
//float nms = .4; //float nms = .4;
net.wait_stream = wait_stream; // 1 - wait CUDA-stream, 0 - not to wait
image im; image im;
im.c = img.c; im.c = img.c;