StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix preprocess orientation

Browse Source
This commit is contained in:
Andre Ubuntu 2023-04-05 20:28:00 -03:00
parent 22bcc7be42
commit 52a8f286ef
1 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/textual_inversion/preprocess.py
View File

@ -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