diff --git a/python/paddle_serving_app/reader/image_reader.py b/python/paddle_serving_app/reader/image_reader.py index 096f46549af137cb04a87e26a3b28c8d42e33daa..6fa6a4e7739d6b7654785c07485c75fc42bd35e6 100644 --- a/python/paddle_serving_app/reader/image_reader.py +++ b/python/paddle_serving_app/reader/image_reader.py @@ -517,6 +517,19 @@ class BGR2RGB(object): return self.__class__.__name__ + "()" +class String2Image(object): + def __init__(self): + pass + + def __call__(self, img_buffer): + data = np.fromstring(img_buffer, np.uint8) + img = cv2.imdecode(data, cv2.IMREAD_COLOR) + return img + + def __repr__(self): + return self.__class__.__name__ + "()" + + class File2Image(object): def __init__(self): pass @@ -561,7 +574,9 @@ class Base64ToImage(object): pass def __call__(self, img_base64): - img = base64.b64decode(img_base64) + sample = base64.b64decode(img_base64) + data = np.fromstring(sample, np.uint8) + img = cv2.imdecode(data, cv2.IMREAD_COLOR) return img def __repr__(self):