From aeb15b3cb9157f5d0b2a9962e17de22560b8a1b2 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sun, 07 Jan 2018 22:47:50 +0000
Subject: [PATCH] Detection and Tracking work well

---
 src/yolo_v2_class.hpp |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/src/yolo_v2_class.hpp b/src/yolo_v2_class.hpp
index 392b71a..5afb22f 100644
--- a/src/yolo_v2_class.hpp
+++ b/src/yolo_v2_class.hpp
@@ -50,6 +50,7 @@
 	const int cur_gpu_id;
 public:
 	float nms = .4;
+	bool wait_stream;
 
 	YOLODLL_API Detector(std::string cfg_filename, std::string weight_filename, int gpu_id = 0);
 	YOLODLL_API ~Detector();
@@ -172,7 +173,7 @@
 		sync_PyrLKOpticalFlow_gpu = cv::cuda::SparsePyrLKOpticalFlow::create();
 		sync_PyrLKOpticalFlow_gpu->setWinSize(cv::Size(21, 21));	// 15, 21, 31
 		sync_PyrLKOpticalFlow_gpu->setMaxLevel(3);		// +- 5 ptx
-		sync_PyrLKOpticalFlow_gpu->setNumIters(1000);	// def: 30
+		sync_PyrLKOpticalFlow_gpu->setNumIters(2000);	// def: 30
 
 		cv::cuda::setDevice(old_gpu_id);
 	}

--
Gitblit v1.10.0