Skip to content

Commit

Permalink
Remove redundant func in json_results (PaddlePaddle#3284)
Browse files Browse the repository at this point in the history
  • Loading branch information
yghstill authored Jun 4, 2021
1 parent 12ae16e commit 32abf1a
Showing 1 changed file with 1 addition and 25 deletions.
26 changes: 1 addition & 25 deletions ppdet/metrics/json_results.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 32abf1a

Please sign in to comment.