From 3558d57681d4ebb235e797b741ad03fb1090f968 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Sun, 9 Oct 2022 14:15:47 +0800 Subject: [PATCH] fix ci --- tools/infer/predict_det.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index fdfe5e57..1b4446a6 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -192,6 +192,8 @@ class TextDetector(object): img_height, img_width = image_shape[0:2] dt_boxes_new = [] for box in dt_boxes: + if type(box) is list: + box = np.array(box) box = self.order_points_clockwise(box) box = self.clip_det_res(box, img_height, img_width) rect_width = int(np.linalg.norm(box[0] - box[1])) -- GitLab