diff options
| author | Andre Ubuntu <goulartnogueira@gmail.com> | 2023-04-05 20:51:29 -0300 |
|---|---|---|
| committer | Andre Ubuntu <goulartnogueira@gmail.com> | 2023-04-05 20:51:29 -0300 |
| commit | 48c06af8dc718abf0bf9355ea5548c6a66e0b1e6 (patch) | |
| tree | 06985c678bd5dadd98b557d617ce91dc78607ac4 | |
| parent | 52a8f286ef99bb5004bea2b099a7bcbb073b638f (diff) | |
Pythonic way to achieve it
| -rw-r--r-- | modules/textual_inversion/preprocess.py | 18 |
1 files changed, 1 insertions, 17 deletions
diff --git a/modules/textual_inversion/preprocess.py b/modules/textual_inversion/preprocess.py index 9cb98694..de1ddb59 100644 --- a/modules/textual_inversion/preprocess.py +++ b/modules/textual_inversion/preprocess.py @@ -162,23 +162,7 @@ def preprocess_work(process_src, process_dst, process_width, process_height, pre filename = os.path.join(src, imagefile)
try:
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 = ImageOps.exif_transpose(img)
img = img.convert("RGB")
except Exception:
continue
|