fix bug

2022-10-10 10:12:28 +08:00 · 2022-10-10 10:12:28 +08:00 · 3628ac1c6c
parent 3558d57681
commit 3628ac1c6c
1 changed files with 2 additions and 2 deletions
--- 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: