OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed wait_stream flag, for compilation without GPU

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
Show Stats 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);
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;