From 72f93cf977ffed232e85c1736ba4355c7afb3fd7 Mon Sep 17 00:00:00 2001 From: wangjingyeye <1025993141@qq.com> Date: Mon, 4 Jul 2022 08:49:34 +0000 Subject: [PATCH] add polygons --- ppocr/postprocess/db_postprocess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/postprocess/db_postprocess.py b/ppocr/postprocess/db_postprocess.py index c61534d6db..e1041b00f9 100755 --- a/ppocr/postprocess/db_postprocess.py +++ b/ppocr/postprocess/db_postprocess.py @@ -236,7 +236,7 @@ class DBPostProcess(object): self.dilation_kernel) else: mask = segmentation[batch_index] - if self.use_polygon: + if self.use_polygon is True: boxes, scores = self.polygons_from_bitmap(pred[batch_index], mask, src_w, src_h) else: