提交 b1f29a38 编写于 作者: M MRXLT

refine ocr reader

上级 cd7dc5bd
...@@ -19,6 +19,7 @@ import math ...@@ -19,6 +19,7 @@ import math
import re import re
import sys import sys
import argparse import argparse
from paddle_serving_app.reader import Sequential, Resize, Transpose, Div, Normalize
class CharacterOps(object): class CharacterOps(object):
...@@ -153,13 +154,15 @@ class OCRReader(object): ...@@ -153,13 +154,15 @@ class OCRReader(object):
resized_w = imgW resized_w = imgW
else: else:
resized_w = int(math.ceil(imgH * ratio)) resized_w = int(math.ceil(imgH * ratio))
resized_image = cv2.resize(img, (resized_w, imgH))
resized_image = resized_image.astype('float32') seq = Sequential([
resized_image = resized_image.transpose((2, 0, 1)) / 255 Resize(imgH, resized_w), Transpose((2, 0, 1)), Div(255),
resized_image -= 0.5 Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5], True)
resized_image /= 0.5 ])
resized_image = seq(img)
padding_im = np.zeros((imgC, imgH, imgW), dtype=np.float32) padding_im = np.zeros((imgC, imgH, imgW), dtype=np.float32)
padding_im[:, :, 0:resized_w] = resized_image padding_im[:, :, 0:resized_w] = resized_image
return padding_im return padding_im
def preprocess(self, img_list): def preprocess(self, img_list):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册