diff --git a/src/coco_kernels.cu b/src/coco_kernels.cu index 4dc3e88f..298bc34a 100644 --- a/src/coco_kernels.cu +++ b/src/coco_kernels.cu @@ -32,7 +32,7 @@ static image det_s; static image disp ; static cv::VideoCapture cap; static float fps = 0; -static int demo_thresh = 0; +static float demo_thresh = 0; void *fetch_in_thread_coco(void *ptr) { diff --git a/src/yolo_kernels.cu b/src/yolo_kernels.cu index e603caf9..d7f1b260 100644 --- a/src/yolo_kernels.cu +++ b/src/yolo_kernels.cu @@ -33,7 +33,7 @@ static image det_s; static image disp ; static cv::VideoCapture cap; static float fps = 0; -static int demo_thresh = 0; +static float demo_thresh = 0; void *fetch_in_thread(void *ptr) {