tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

save as png

Browse Source
This commit is contained in:
Jonas Konschak 2022-07-15 01:24:04 +02:00 committed by GitHub
parent 71fb0c29bc
commit 1ab70c790b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
image_from_text.py
View File

@ -27,9 +27,9 @@ def ascii_from_image(image: Image.Image, size: int = 128) -> str:
def save_image(image: Image.Image, path: str): def save_image(image: Image.Image, path: str):
if os.path.isdir(path): if os.path.isdir(path):
path = os.path.join(path, 'generated.jpg') path = os.path.join(path, 'generated.png')
elif not path.endswith('.jpg'): elif not path.endswith('.png'):
path += '.jpg' path += '.png'
print("saving image to", path) print("saving image to", path)
image.save(path) image.save(path)
return image return image
@ -76,4 +76,4 @@ if __name__ == '__main__':
image_path=args.image_path, image_path=args.image_path,
models_root=args.models_root, models_root=args.models_root,
fp16=args.fp16, fp16=args.fp16,
) )