From 37259078f02b797ece6bf67aa6179cb05ba92d47 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Sat, 05 Aug 2017 13:23:53 +0000
Subject: [PATCH] Add tracking for image in example of usage SO/DLL
---
src/yolo_console_dll.cpp | 7 ++++++-
1 files changed, 6 insertions(+), 1 deletions(-)
diff --git a/src/yolo_console_dll.cpp b/src/yolo_console_dll.cpp
index e70f1de..e9c6abe 100644
--- a/src/yolo_console_dll.cpp
+++ b/src/yolo_console_dll.cpp
@@ -97,13 +97,18 @@
else
for (std::string line; file >> line;) {
std::cout << line << std::endl;
- show_result(detector.detect(cv::imread(line)), obj_names);
+ cv::Mat mat_img = cv::imread(line);
+ std::vector<bbox_t> result_vec = detector.detect(mat_img);
+ show_result(result_vec, obj_names);
+ //draw_boxes(mat_img, result_vec, obj_names);
+ //cv::imwrite("res_" + line, mat_img);
}
}
else { // image file
cv::Mat mat_img = cv::imread(filename);
std::vector<bbox_t> result_vec = detector.detect(mat_img);
+ result_vec = detector.tracking(result_vec); // comment it - if track_id is not required
draw_boxes(mat_img, result_vec, obj_names);
show_result(result_vec, obj_names);
}
--
Gitblit v1.10.0