From 3628ac1c6c0fad4e12cf8525a3072f1f3edbd933 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 10 Oct 2022 10:12:28 +0800 Subject: [PATCH] fix bug --- ppocr/postprocess/db_postprocess.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppocr/postprocess/db_postprocess.py b/ppocr/postprocess/db_postprocess.py index cbe7bf4b..dfe10781 100755 --- a/ppocr/postprocess/db_postprocess.py +++ b/ppocr/postprocess/db_postprocess.py @@ -233,10 +233,10 @@ class DBPostProcess(object): self.dilation_kernel) else: mask = segmentation[batch_index] - if self.box_type == 'quad': + if self.box_type == 'poly': boxes, scores = self.polygons_from_bitmap(pred[batch_index], mask, src_w, src_h) - elif self.box_type == 'poly': + elif self.box_type == 'quad': boxes, scores = self.boxes_from_bitmap(pred[batch_index], mask, src_w, src_h) else: -- GitLab