From edc0fd0ccd93027add6e119467dd3c8ec6b86500 Mon Sep 17 00:00:00 2001 From: MissPenguin Date: Sun, 13 Dec 2020 06:18:26 +0000 Subject: [PATCH] fix predict_det --- ppocr/postprocess/db_postprocess.py | 2 +- tools/infer/predict_det.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) mode change 100644 => 100755 ppocr/postprocess/db_postprocess.py diff --git a/ppocr/postprocess/db_postprocess.py b/ppocr/postprocess/db_postprocess.py old mode 100644 new mode 100755 index 0be2c12a..16c789dc --- a/ppocr/postprocess/db_postprocess.py +++ b/ppocr/postprocess/db_postprocess.py @@ -40,7 +40,7 @@ class DBPostProcess(object): self.max_candidates = max_candidates self.unclip_ratio = unclip_ratio self.min_size = 3 - self.dilation_kernel = None if not use_dilation else [[1, 1], [1, 1]] + self.dilation_kernel = None if not use_dilation else np.array([[1, 1], [1, 1]]) def boxes_from_bitmap(self, pred, _bitmap, dest_width, dest_height): ''' diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 43db20d2..6f98ded8 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -63,6 +63,7 @@ class TextDetector(object): postprocess_params["box_thresh"] = args.det_db_box_thresh postprocess_params["max_candidates"] = 1000 postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio + postprocess_params["use_dilation"] = True else: logger.info("unknown det_algorithm:{}".format(self.det_algorithm)) sys.exit(0) @@ -111,7 +112,7 @@ class TextDetector(object): box = self.clip_det_res(box, img_height, img_width) rect_width = int(np.linalg.norm(box[0] - box[1])) 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 dt_boxes_new.append(box) dt_boxes = np.array(dt_boxes_new) -- GitLab