From 57fce97fd5c2486997a6ee9ac67cccf2a5f35c81 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Wed, 18 Apr 2018 22:17:24 +0300 Subject: [PATCH] yolo_console_dll.cpp uses Yolo v3 by default --- src/yolo_console_dll.cpp | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/yolo_console_dll.cpp b/src/yolo_console_dll.cpp index 2e26a505..ea4330e3 100644 --- a/src/yolo_console_dll.cpp +++ b/src/yolo_console_dll.cpp @@ -226,9 +226,9 @@ std::vector objects_names_from_file(std::string const filename) { int main(int argc, char *argv[]) { - std::string names_file = "data/voc.names"; - std::string cfg_file = "cfg/yolo-voc.cfg"; - std::string weights_file = "yolo-voc.weights"; + std::string names_file = "data/coco.names"; + std::string cfg_file = "cfg/yolov3.cfg"; + std::string weights_file = "yolov3.weights"; std::string filename; if (argc > 4) { //voc.names yolo-voc.cfg yolo-voc.weights test.mp4 @@ -457,7 +457,13 @@ int main(int argc, char *argv[]) } else { // image file cv::Mat mat_img = cv::imread(filename); + + auto start = std::chrono::steady_clock::now(); std::vector result_vec = detector.detect(mat_img); + auto end = std::chrono::steady_clock::now(); + std::chrono::duration spent = end - start; + std::cout << " Time: " << spent.count() << " sec \n"; + //result_vec = detector.tracking_id(result_vec); // comment it - if track_id is not required draw_boxes(mat_img, result_vec, obj_names); cv::imshow("window name", mat_img);