fix automatic orientation phone

app.py

@@ -3,11 +3,29 @@ from typing import Tuple
 
 import gradio as gr
 import numpy as np
-from PIL import Image
+from PIL import Image, ExifTags
 
 from Segmentation.segmentation import get_mask, replace_sofa
 from StyleTransfer.styleTransfer import create_styledSofa
 
+def fix_orient(img: Image.Image) -> Image.Image:
+    for orientation in ExifTags.TAGS.keys():
+        if ExifTags.TAGS[orientation]=='Orientation':
+            break
+    info=img._getexif()
+    if (info):
+        info = dict(info.items())
+        orientation = info[orientation]
+        if (orientation == 1) | (orientation == 2):
+            img = img
+        if (orientation == 3) | (orientation == 4):
+            img = img.rotate(180,expand=True)
+        if (orientation == 5) | (orientation == 6):
+            img = img.rotate(270,expand=True)
+        if (orientation == 7) | (orientation == 8):
+            img = img.rotate(90,expand=True)
+    return img
+
 
 def resize_sofa(img: Image.Image) -> Tuple[Image.Image, tuple]:
     """
@@ -105,8 +123,8 @@ def style_sofa(
     print("Starting job ", id)
     # preprocess input images to fit requirements of the segmentation model
     input_img, style_img = Image.fromarray(Input_image), Image.fromarray(Style_image)
-    resized_img, box = resize_sofa(input_img)
-    resized_style = resize_style(style_img)
+    resized_img, box = resize_sofa(fix_orient(input_img))
+    resized_style = resize_style(fix_orient(style_img))
     # resized_style.save('resized_style.jpg')
     # generate mask for image
     print("generating mask...")
@@ -177,10 +195,7 @@ demo = gr.Interface(
     target='_blank'> \
     4. A tensorflow model for fast arbitrary image style transfer. \
     </a>  \n"
-    "<a href='https://github.com/PaddlePaddle/PaddleHub/tree/release/v2.2/modules/image/Image_gan/style_transfer/stylepro_artistic' \
-    target='_blank'> \
-    5. A paddleHub model for parameter free style transfer. \
-    </a>  \n",
+    "<a href='https://github.com/PaddlePaddle/PaddleHub/tree/release/v2.2/modules/image/Image_gan/style_transfer/stylepro_artistic' target='_blank'> 5. A paddleHub model for parameter free style transfer.  </a>  \n", # noqa
 )
 
 if __name__ == "__main__":