From 0a8391e816be243f2d75b643d12778a0c5295455 Mon Sep 17 00:00:00 2001
From: AlexeyAB <kikots@mail.ru>
Date: Sun, 06 Aug 2017 19:21:46 +0000
Subject: [PATCH] Added to Makefile the path to include of cuDNN
---
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