fix ci

truongnn1106 · Oct 9, 2022 · 3558d57 · 3558d57
1 parent 119196f
commit 3558d57
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
@@ -192,6 +192,8 @@ def filter_tag_det_res(self, dt_boxes, image_shape):
  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]))