support img_h < 32 (#1992)

4 years ago · 3ab2d46310
parent 0b53465894
commit 3ab2d46310
1 changed files with 2 additions and 2 deletions
--- a/ppocr/data/imaug/operators.py
+++ b/ppocr/data/imaug/operators.py
@ -185,8 +185,8 @@ class DetResizeForTest(object):
        resize_h = int(h * ratio)
        resize_w = int(w * ratio)

-        resize_h = int(round(resize_h / 32) * 32)
-        resize_w = int(round(resize_w / 32) * 32)
+        resize_h = max(int(round(resize_h / 32) * 32), 32)
+        resize_w = max(int(round(resize_w / 32) * 32), 32)

        try:
            if int(resize_w) <= 0 or int(resize_h) <= 0: