From 2103753d929387715ba8c20b3397c4b1903103cc Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 10 Oct 2022 10:54:02 +0800 Subject: [PATCH] sorted --- tools/infer/predict_system.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/infer/predict_system.py b/tools/infer/predict_system.py index e0f2c41f..f12fd626 100755 --- a/tools/infer/predict_system.py +++ b/tools/infer/predict_system.py @@ -116,7 +116,10 @@ def sorted_boxes(dt_boxes): sorted boxes(array) with shape [4, 2] """ num_boxes = dt_boxes.shape[0] - sorted_boxes = sorted(dt_boxes, key=lambda x: (x[0][1], x[0][0])) + if abs(num_boxes - 2) < 1e-4: + sorted_boxes = sorted(dt_boxes, key=lambda x: (x[1], x[0])) + else: + sorted_boxes = sorted(dt_boxes, key=lambda x: (x[0][1], x[0][0])) _boxes = list(sorted_boxes) for i in range(num_boxes - 1): -- GitLab