From b296933ba01c8a7949fd06f7473cc832d06dd9fe Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Fri, 7 Oct 2022 12:05:33 -0400 Subject: [PATCH] autorotate init images using exif orientation tag --- ldm/generate.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ldm/generate.py b/ldm/generate.py index 292329d24c..37e84f05e6 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -849,6 +849,7 @@ class Generate: print( f'>> loaded input image of size {image.width}x{image.height}' ) + image = ImageOps.exif_transpose(image) return image def _create_init_image(self, image, width, height, fit=True): @@ -857,7 +858,6 @@ class Generate: image = self._fit_image(image, (width, height)) else: image = self._squeeze_image(image) - image = np.array(image).astype(np.float32) / 255.0 image = image[None].transpose(0, 3, 1, 2) image = torch.from_numpy(image) @@ -874,7 +874,6 @@ class Generate: image = self._fit_image(image, (width, height)) else: image = self._squeeze_image(image) - image = image.resize((image.width//downsampling, image.height // downsampling), resample=Image.Resampling.NEAREST) image = np.array(image)