From 32abf1a06fc80258aa5bc0c5be206da9b54bd896 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 4 Jun 2021 21:00:08 +0800
Subject: [PATCH] Remove redundant func in json_results (#3284)

---
 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