| | |
| | | YOLODLL_API int get_net_width(); |
| | | YOLODLL_API int get_net_height(); |
| | | |
| | | YOLODLL_API std::vector<bbox_t> tracking(std::vector<bbox_t> cur_bbox_vec, int const frames_story = 4); |
| | | YOLODLL_API std::vector<bbox_t> tracking(std::vector<bbox_t> cur_bbox_vec, int const frames_story = 6); |
| | | |
| | | #ifdef OPENCV |
| | | std::vector<bbox_t> detect(cv::Mat mat, float thresh = 0.2, bool use_mean = false) |
| | |
| | | cv::Mat det_mat; |
| | | cv::resize(mat, det_mat, cv::Size(get_net_width(), get_net_height())); |
| | | auto image_ptr = mat_to_image(det_mat); |
| | | auto detection_boxes = detect(*image_ptr, thresh); |
| | | auto detection_boxes = detect(*image_ptr, thresh, use_mean); |
| | | float wk = (float)mat.cols / det_mat.cols, hk = (float)mat.rows / det_mat.rows; |
| | | for (auto &i : detection_boxes) i.x*=wk, i.w*= wk, i.y*=hk, i.h*=hk; |
| | | return detection_boxes; |