From 3659d84f24ddc95102483cca430e01dc05568cbb Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Fri, 31 Mar 2017 18:42:51 +0000
Subject: [PATCH] Added tracking: numerating the detected objects on video

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

diff --git a/src/yolo_v2_class.cpp b/src/yolo_v2_class.cpp
index ea13ea3..8643a22 100644
--- a/src/yolo_v2_class.cpp
+++ b/src/yolo_v2_class.cpp
@@ -154,7 +154,7 @@
 	cudaSetDevice(net.gpu_index);
 	//std::cout << "net.gpu_index = " << net.gpu_index << std::endl;
 
-	float nms = .4;
+	//float nms = .4;
 
 	image im;
 	im.c = img.c;
@@ -189,6 +189,7 @@
 			bbox.h = b.h*im.h;
 			bbox.obj_id = obj_id;
 			bbox.prob = prob;
+			bbox.track_id = 0;
 
 			bbox_vec.push_back(bbox);
 		}

--
Gitblit v1.10.0