sontungdo/sd-image-editor
1
0
Fork 0
mirror of https://github.com/sontungdo/sd-image-editor.git synced 2026-01-26 19:29:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle when image is None

Browse Source
This commit is contained in:
w-e-w
2024-04-05 14:57:37 +09:00
parent 9c5d58fae7
commit 3e0afbb3a2
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/sd-image-editor.py
View File

@@ -13,6 +13,8 @@ from PIL import Image, ImageEnhance, ImageFilter, ImageTransform
save_path = os.path.join("output", "img2img-images", "sd-image-editor")
def edit(img, degree, expand, flip, interpolate_mode, color, contrast, brightness, sharpness):
if img is None:
return None
# Flip
if flip:
img = img.transpose(method=Image.Transpose.FLIP_LEFT_RIGHT)