fix bugs for SLANet infer (#13861)

2024-09-13 12:53:09 +08:00 · 2024-09-13 12:53:09 +08:00 · ac5313d0b1
parent 284a20bf10
commit ac5313d0b1
1 changed files with 3 additions and 0 deletions
--- a/ppocr/postprocess/table_postprocess.py
+++ b/ppocr/postprocess/table_postprocess.py
@ -129,8 +129,11 @@ class TableLabelDecode(AttnLabelDecode):

    def _bbox_decode(self, bbox, shape):
        h, w, ratio_h, ratio_w, pad_h, pad_w = shape
+        h, w = pad_h, pad_w
        bbox[0::2] *= w
        bbox[1::2] *= h
+        bbox[0::2] /= ratio_w
+        bbox[1::2] /= ratio_h
        return bbox