Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf(invoke_ai_web_server): encode intermediate result previews as jpeg

Browse Source 
For size savings of about 80%, and jpeg encoding is still plenty fast.
This commit is contained in:
Kevin Turner 2023-02-25 20:59:03 -08:00
parent 6696882c71
commit a3076cf951
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
invokeai/backend/invoke_ai_web_server.py
View File

@ -1062,7 +1062,7 @@ class InvokeAIWebServer:
(width, height) = image.size
width *= 8
height *= 8
img_base64 = image_to_dataURL(image)
img_base64 = image_to_dataURL(image, image_format="JPEG")
self.socketio.emit(
"intermediateResult",
{
@ -1714,13 +1714,14 @@ def dataURL_to_image(dataURL: str) -> ImageType:
return image
def image_to_dataURL(image: ImageType) -> str:
def image_to_dataURL(image: ImageType, image_format:str="PNG") -> str:
"""
Converts an image into a base64 image dataURL.
"""
buffered = io.BytesIO()
image.save(buffered, format="PNG")
image_base64 = "data:image/png;base64," + base64.b64encode(
image.save(buffered, format=image_format)
mime_type = Image.MIME.get(image_format.upper(), "image/" + image_format.lower())
image_base64 = f"data:{mime_type};base64," + base64.b64encode(
buffered.getvalue()
).decode("UTF-8")
return image_base64