From f4a30ef60a3c9288967386d3b023fa3cf7ffb465 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Tue, 08 Aug 2017 19:16:28 +0000
Subject: [PATCH] Track id fixed
---
src/yolo_console_dll.cpp | 13 ++++++++-----
1 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/src/yolo_console_dll.cpp b/src/yolo_console_dll.cpp
index cada95e..e1f2c7e 100644
--- a/src/yolo_console_dll.cpp
+++ b/src/yolo_console_dll.cpp
@@ -63,17 +63,19 @@
}
-int main()
+int main(int argc, char *argv[])
{
+ std::string filename;
+ if (argc > 1) filename = argv[1];
+
Detector detector("yolo-voc.cfg", "yolo-voc.weights");
auto obj_names = objects_names_from_file("data/voc.names");
while (true)
- {
- std::string filename;
+ {
std::cout << "input image or video filename: ";
- std::cin >> filename;
+ if(filename.size() == 0) std::cin >> filename;
if (filename.size() == 0) break;
try {
@@ -94,7 +96,7 @@
td.join();
result_vec = thread_result_vec;
det_frame = frame;
- td = std::thread([&]() { thread_result_vec = detector.detect(det_frame, 0.2, true); ready_flag = true; });
+ td = std::thread([&]() { thread_result_vec = detector.detect(det_frame, 0.24, true); ready_flag = true; });
if (!prev_frame.empty()) {
result_vec = detector.tracking(result_vec); // comment it - if track_id is not required
@@ -138,6 +140,7 @@
}
catch (std::exception &e) { std::cerr << "exception: " << e.what() << "\n"; getchar(); }
catch (...) { std::cerr << "unknown exception \n"; getchar(); }
+ filename.clear();
}
return 0;
--
Gitblit v1.10.0