soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Pythonic way to achieve it

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

18
modules/textual_inversion/preprocess.py
View File

@ -162,23 +162,7 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre
filename = os.path.join(src, imagefile) filename = os.path.join(src, imagefile)
try: try:
img = Image.open(filename) img = Image.open(filename)
# make sure to rotate the image according to EXIF data of the original image img = ImageOps.exif_transpose(img)
# 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") img = img.convert("RGB")
except Exception: except Exception:
continue continue