AlexeyAB
2017-08-05 37259078f02b797ece6bf67aa6179cb05ba92d47
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);
         }