diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-10 23:26:24 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-10 23:26:24 +0100 |
| commit | e0fbe6d27e7b4505766c8cb5a4264e1114cf3721 (patch) | |
| tree | a0a997a36b1b7fc100abc87e31e973440580a824 /modules/textual_inversion | |
| parent | 767202a4c324f9b49f63ab4dabbb5736fe9df6e5 (diff) | |
colour depth conversion fix
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 f3cacaa0..ae807268 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -103,7 +103,7 @@ def crop_black(img,tol=0): def extractImageDataEmbed(image):
d=3
- outarr = crop_black(np.array(image.getdata()).reshape(image.size[1],image.size[0],d ).astype(np.uint8) ) & 0x0F
+ outarr = crop_black(np.array(image.convert('RGB').getdata()).reshape(image.size[1],image.size[0],d ).astype(np.uint8) ) & 0x0F
blackCols = np.where( np.sum(outarr, axis=(0,2))==0)
if blackCols[0].shape[0] < 2:
print('No Image data blocks found.')
|