Commit 52a8f286 authored by Andre Ubuntu's avatar Andre Ubuntu
Browse files

fix preprocess orientation

parent 22bcc7be
Loading
Loading
Loading
Loading
+19 −1
Original line number Diff line number Diff line
@@ -161,7 +161,25 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
        params.subindex = 0
        filename = os.path.join(src, imagefile)
        try:
            img = Image.open(filename).convert("RGB")
            img = Image.open(filename)
            # make sure to rotate the image according to EXIF data of the original image
            # ImageOps.exif_transpose(img) # doesn't work for some reason
            EXIF = img._getexif()
            # rotate the image according to the EXIF data
            try:
                if EXIF[274] == 3:
                    # print("Rotating image by 180 degrees")
                    img = img.rotate(180, expand=True)
                elif EXIF[274] == 6:
                    # print("Rotating image by 270 degrees")
                    img = img.rotate(270, expand=True)
                elif EXIF[274] == 8:
                    # print("Rotating image by 90 degrees")
                    img = img.rotate(90, expand=True)
            except:
                pass
                # print("No EXIF data found for image: " + filename)
            img = img.convert("RGB")
        except Exception:
            continue