| | |
| | | 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); |
| | | } |