handle exif data when loading images during training (#924)

v7labs · Sep 3, 2024 · c0888e1 · c0888e1
1 parent 3b23870
commit c0888e1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/darwin/dataset/utils.py b/darwin/dataset/utils.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 from PIL import Image as PILImage
+from PIL import ImageOps
 from rich.live import Live
 from rich.progress import ProgressBar, track
 
@@ -676,6 +677,7 @@ def load_pil_image(path: Path, to_rgb: Optional[bool] = True) -> PILImage.Image:
         The loaded image.
     """
     pic = PILImage.open(path)
+    pic = ImageOps.exif_transpose(pic)
     if to_rgb:
         pic = convert_to_rgb(pic)
     return pic