diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-10 15:43:09 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-10 15:43:09 +0100 |
| commit | df6d0d9286279c41c4c67460c3158fa268697524 (patch) | |
| tree | cf3e1b24bbe8a97d7268af5e053ef13925b5cca6 /modules/textual_inversion | |
| parent | 707a431100362645e914042bb344d08439f48ac8 (diff) | |
convert back to rgb as some hosts add alpha
Diffstat (limited to 'modules/textual_inversion')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 6fb64691..667a7cf2 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -64,7 +64,7 @@ def crop_black(img,tol=0): def extractImageDataFooter(image):
d=3
- outarr = crop_black(np.array(image.getdata()).reshape(image.size[1],image.size[0],d ).astype(np.uint8) )
+ outarr = crop_black(np.array(image.convert('RGB').getdata()).reshape(image.size[1],image.size[0],d ).astype(np.uint8) )
lastRow = np.where( np.sum(outarr, axis=(1,2))==0)
if lastRow[0].shape[0] == 0:
print('Image data block not found.')
|