michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add file based test

Browse source
This commit is contained in:
DepFA 2022-10-11 20:21:30 +01:00 committed by GitHub
parent 7e6a6e00ad
commit 66ec505975
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
modules/textual_inversion/image_embedding.py
View file

@ -164,6 +164,14 @@ def caption_image_overlay(srcimage,title,footerLeft,footerMid,footerRight,textfo
return image return image
if __name__ == '__main__': if __name__ == '__main__':
testEmbed = Image.open('test_embedding.png')
data = extract_image_data_embed(testEmbed)
assert data is not None
data = embedding_from_b64(testEmbed.text['sd-ti-embedding'])
assert data is not None
image = Image.new('RGBA',(512,512),(255,255,200,255)) image = Image.new('RGBA',(512,512),(255,255,200,255))
cap_image = caption_image_overlay(image, 'title', 'footerLeft', 'footerMid', 'footerRight') cap_image = caption_image_overlay(image, 'title', 'footerLeft', 'footerMid', 'footerRight')