File2Image support py3

5176e33f · MRXLT · 2ae4ffba · 5176e33f
显示空白变更内容
内联并排

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):
+        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)