Searched refs:loc_data (Results 1 – 2 of 2) sorted by relevance
45 def ssd_post_process(conf_data, loc_data): argument48 prior_bboxes = prior_data[:len(loc_data)]49 prior_variances = prior_data[len(loc_data):]51 prior_num = int(len(loc_data) / 4) # 873274 bbox_center_x = prior_variances[4*i+0] * loc_data[4*i+0] * prior_w + prior_center_x75 bbox_center_y = prior_variances[4*i+1] * loc_data[4*i+1] * prior_h + prior_center_y76 bbox_w = math.exp(prior_variances[4*i+2] * loc_data[4*i+2]) * prior_w77 bbox_h = math.exp(prior_variances[4*i+3] * loc_data[4*i+3]) * prior_h
70 def ssd_post_process(conf_data, loc_data, imgpath, output_dir='.'): argument71 prior_num = int(len(loc_data) / 4) # num prior boxes98 bbox_center_x = loc_data[4 * i + 1] / X_SCALE * prior_bboxes[3][i] + prior_bboxes[1][i]99 bbox_center_y = loc_data[4 * i + 0] / Y_SCALE * prior_bboxes[2][i] + prior_bboxes[0][i]100 bbox_w = math.exp(loc_data[4 * i + 3] / W_SCALE) * prior_bboxes[3][i]101 bbox_h = math.exp(loc_data[4 * i + 2] / H_SCALE) * prior_bboxes[2][i]