diff --git a/deploy/python/infer.py b/deploy/python/infer.py index 0ab948904b90ad1a38cf957d1f34bbffe7fd96e6..a6ab42c4a5c5a23504271637dbf69699fbc32d0c 100644 --- a/deploy/python/infer.py +++ b/deploy/python/infer.py @@ -341,7 +341,7 @@ class Config(): def print_config(self): print('----------- Model Configuration -----------') print('%s: %s' % ('Model Arch', self.arch)) - print('%s: %s' % ('Use Padddle Executor', self.use_python_inference)) + print('%s: %s' % ('Use Paddle Executor', self.use_python_inference)) print('%s: ' % ('Transform Order')) for op_info in self.preprocess_infos: print('--%s: %s' % ('transform op', op_info['type'])) diff --git a/tools/infer.py b/tools/infer.py index 99a45b09ef43a62f71ee3b4bf1b145bb877ecb99..bd26650bbec56e890507927874ec93a9375851c2 100644 --- a/tools/infer.py +++ b/tools/infer.py @@ -25,7 +25,7 @@ if parent_path not in sys.path: import glob import numpy as np import six -from PIL import Image +from PIL import Image, ImageOps import paddle from paddle import fluid @@ -205,6 +205,7 @@ def main(): for im_id in im_ids: image_path = imid2path[int(im_id)] image = Image.open(image_path).convert('RGB') + image = ImageOps.exif_transpose(image) # use VisualDL to log original image if FLAGS.use_vdl: