Lines Matching refs:dis_pred
117 static void generate_proposals(const ncnn::Mat& cls_pred, const ncnn::Mat& dis_pred, int stride, co… in generate_proposals() argument
135 const int reg_max_1 = dis_pred.w / 4; in generate_proposals()
159 ncnn::Mat bbox_pred(reg_max_1, 4, (void*)dis_pred.row(idx)); in generate_proposals()
181 float dis_pred[4]; in generate_proposals() local
191 dis_pred[k] = dis * stride; in generate_proposals()
197 float x0 = pb_cx - dis_pred[0]; in generate_proposals()
198 float y0 = pb_cy - dis_pred[1]; in generate_proposals()
199 float x1 = pb_cx + dis_pred[2]; in generate_proposals()
200 float y1 = pb_cy + dis_pred[3]; in generate_proposals()
273 ncnn::Mat dis_pred; in detect_nanodet() local
275 ex.extract("795", dis_pred); in detect_nanodet()
278 generate_proposals(cls_pred, dis_pred, 8, in_pad, prob_threshold, objects8); in detect_nanodet()
286 ncnn::Mat dis_pred; in detect_nanodet() local
288 ex.extract("817", dis_pred); in detect_nanodet()
291 generate_proposals(cls_pred, dis_pred, 16, in_pad, prob_threshold, objects16); in detect_nanodet()
299 ncnn::Mat dis_pred; in detect_nanodet() local
301 ex.extract("839", dis_pred); in detect_nanodet()
304 generate_proposals(cls_pred, dis_pred, 32, in_pad, prob_threshold, objects32); in detect_nanodet()