diff --git a/ppocr/postprocess/db_postprocess.py b/ppocr/postprocess/db_postprocess.py index cbe7bf4bd6ba33e13c0b0718ff2a585014ab6acc..dfe107816c195b36bf06568843b008bf66ff24c7 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: