fix predict det
This commit is contained in:
parent
bc7dd36e0e
commit
954c3313d2
|
@ -117,7 +117,7 @@ class TextDetector(object):
|
||||||
box = self.clip_det_res(box, img_height, img_width)
|
box = self.clip_det_res(box, img_height, img_width)
|
||||||
rect_width = int(np.linalg.norm(box[0] - box[1]))
|
rect_width = int(np.linalg.norm(box[0] - box[1]))
|
||||||
rect_height = int(np.linalg.norm(box[0] - box[3]))
|
rect_height = int(np.linalg.norm(box[0] - box[3]))
|
||||||
if rect_width <= 10 or rect_height <= 10:
|
if rect_width <= 3 or rect_height <= 3:
|
||||||
continue
|
continue
|
||||||
dt_boxes_new.append(box)
|
dt_boxes_new.append(box)
|
||||||
dt_boxes = np.array(dt_boxes_new)
|
dt_boxes = np.array(dt_boxes_new)
|
||||||
|
|
Loading…
Reference in New Issue