mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

add polygons

Browse Source
This commit is contained in:
wangjingyeye 2022-07-04 08:49:34 +00:00
parent 8d9cfade49
commit 72f93cf977
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/postprocess/db_postprocess.py
View File

@ -236,7 +236,7 @@ class DBPostProcess(object):
self.dilation_kernel) self.dilation_kernel)
else: else:
mask = segmentation[batch_index] mask = segmentation[batch_index]
if self.use_polygon: if self.use_polygon is True:
boxes, scores = self.polygons_from_bitmap(pred[batch_index], boxes, scores = self.polygons_from_bitmap(pred[batch_index],
mask, src_w, src_h) mask, src_w, src_h)
else: else: