From 6ec36068ee450df33c0038fb0cdc575b21510acd Mon Sep 17 00:00:00 2001 From: will-jl944 Date: Fri, 4 Jun 2021 20:21:05 +0800 Subject: [PATCH] remove redundant function; get category_id according to label_to_cat_id_map (#3282) --- ppdet/metrics/json_results.py | 26 +------------------------- 1 file changed, 1 insertion(+), 25 deletions(-) diff --git a/ppdet/metrics/json_results.py b/ppdet/metrics/json_results.py index 7bb509c43..94b3c827f 100755 --- a/ppdet/metrics/json_results.py +++ b/ppdet/metrics/json_results.py @@ -55,31 +55,7 @@ def get_det_poly_res(bboxes, bbox_nums, image_id, label_to_cat_id_map, bias=0): num_id, score, x1, y1, x2, y2, x3, y3, x4, y4 = dt.tolist() if int(num_id) < 0: continue - category_id = int(num_id) - rbox = [x1, y1, x2, y2, x3, y3, x4, y4] - dt_res = { - 'image_id': cur_image_id, - 'category_id': category_id, - 'bbox': rbox, - 'score': score - } - det_res.append(dt_res) - return det_res - - -def get_det_poly_res(bboxes, bbox_nums, image_id, label_to_cat_id_map, bias=0): - det_res = [] - k = 0 - for i in range(len(bbox_nums)): - cur_image_id = int(image_id[i][0]) - det_nums = bbox_nums[i] - for j in range(det_nums): - dt = bboxes[k] - k = k + 1 - num_id, score, x1, y1, x2, y2, x3, y3, x4, y4 = dt.tolist() - if int(num_id) < 0: - continue - category_id = int(num_id) + category_id = label_to_cat_id_map[int(num_id)] rbox = [x1, y1, x2, y2, x3, y3, x4, y4] dt_res = { 'image_id': cur_image_id, -- GitLab