File2Image support py3

325706fc · MRXLT · bd42658a · 325706fc
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

python/paddle_serving_app/reader/image_reader.py python/paddle_serving_app/reader/image_reader.py +4 -1

未找到文件。
--- a/python/paddle_serving_app/reader/image_reader.py
+++ b/python/paddle_serving_app/reader/image_reader.py
@@ -296,7 +296,10 @@ class File2Image(object):
        pass

    def __call__(self, img_path):
-        fin = open(img_path)
+        if py_version == 2:
+            fin = open(img_path)
+        else:
+            fin = open(img_path, "rb")
        sample = fin.read()
        data = np.fromstring(sample, np.uint8)
        img = cv2.imdecode(data, cv2.IMREAD_COLOR)