Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "a7f25b4a881c0835980e2a3db5fe09bfc4ab2dc0"
Unverified Commit 28a40efe authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #3809 from tink2123/fix_infer_shape 

fix infer shape
parents d77db841 22b57d63
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/infer/predict_rec.py
View file @ 28a40efe
...@@ -88,8 +88,8 @@ class TextRecognizer(object): ...@@ -88,8 +88,8 @@ class TextRecognizer(object):
def resize_norm_img(self, img, max_wh_ratio): def resize_norm_img(self, img, max_wh_ratio):
imgC, imgH, imgW = self.rec_image_shape imgC, imgH, imgW = self.rec_image_shape
assert imgC == img.shape[2] assert imgC == img.shape[2]
if self.character_type == "ch": max_wh_ratio = max(max_wh_ratio, imgW / imgH)
imgW = int((32 * max_wh_ratio)) imgW = int((32 * max_wh_ratio))
h, w = img.shape[:2] h, w = img.shape[:2]
ratio = w / float(h) ratio = w / float(h)
if math.ceil(imgH * ratio) > imgW: if math.ceil(imgH * ratio) > imgW:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment