diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-10-11 19:21:30 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-11 19:21:30 +0000 |
| commit | 66ec505975aaa305a217fc27281ce368cbaef281 (patch) | |
| tree | a397f1e5d0e43f4452ea7e9479e959ad4edadb3d /modules/textual_inversion/image_embedding.py | |
| parent | 7e6a6e00ad6f3b7ef43c8120db9ecac6e8d6bea5 (diff) | |
| download | stable-diffusion-webui-gfx803-66ec505975aaa305a217fc27281ce368cbaef281.tar.gz stable-diffusion-webui-gfx803-66ec505975aaa305a217fc27281ce368cbaef281.tar.bz2 stable-diffusion-webui-gfx803-66ec505975aaa305a217fc27281ce368cbaef281.zip | |
add file based test
Diffstat (limited to 'modules/textual_inversion/image_embedding.py')
| -rw-r--r-- | modules/textual_inversion/image_embedding.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/modules/textual_inversion/image_embedding.py b/modules/textual_inversion/image_embedding.py index c67028a5..1224fb42 100644 --- a/modules/textual_inversion/image_embedding.py +++ b/modules/textual_inversion/image_embedding.py @@ -164,6 +164,14 @@ def caption_image_overlay(srcimage,title,footerLeft,footerMid,footerRight,textfo return image
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))
cap_image = caption_image_overlay(image, 'title', 'footerLeft', 'footerMid', 'footerRight')
|