forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix db postprocess

This commit is contained in:
MissPenguin 2020-12-09 08:51:43 +00:00
parent 661e459e3e
commit d42bf7a0b7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppocr/postprocess/db_postprocess.py
View File

@ -138,9 +138,9 @@ class DBPostProcess(object):
boxes_batch = [] boxes_batch = []
for batch_index in range(pred.shape[0]): for batch_index in range(pred.shape[0]):
height, width = shape_list[batch_index] src_h, src_w, ratio_h, ratio_w = shape_list[batch_index]
boxes, scores = self.boxes_from_bitmap( boxes, scores = self.boxes_from_bitmap(
pred[batch_index], segmentation[batch_index], width, height) pred[batch_index], segmentation[batch_index], src_w, src_h)
boxes_batch.append({'points': boxes}) boxes_batch.append({'points': boxes})
return boxes_batch return boxes_batch