revert modified for predict_rec

This commit is contained in:
tink2123 2020-06-30 14:47:24 +08:00
parent 1c8cddf46f
commit 2cc1d00b04

View File

@ -53,7 +53,7 @@ class TextRecognizer(object):
imgC, imgH, imgW = self.rec_image_shape
assert imgC == img.shape[2]
if self.character_type == "ch":
imgW = int(math.ceil(32 * max_wh_ratio))
imgW = int((32 * max_wh_ratio))
h, w = img.shape[:2]
ratio = w / float(h)
if math.ceil(imgH * ratio) > imgW: