| 144 | } |
| 145 | |
| 146 | void process_boxes(std::vector<face_box>& input, int img_h, int img_w, std::vector<face_box>& rects) |
| 147 | { |
| 148 | nms_boxes(input, 0.7, NMS_UNION, rects); |
| 149 | |
| 150 | regress_boxes(rects); |
| 151 | |
| 152 | square_boxes(rects); |
| 153 | |
| 154 | padding(img_h, img_w, rects); |
| 155 | } |
| 156 | |
| 157 | void generate_bounding_box(const float* confidence_data, int confidence_size, const float* reg_data, float scale, |
| 158 | float threshold, int feature_h, int feature_w, std::vector<face_box>& output, |
no test coverage detected