add segmind vega

This commit is contained in:
Vladimir Mandic
2023-12-16 18:06:10 -05:00
parent 86af00ce36
commit b9df9be5a5
7 changed files with 20 additions and 4 deletions
+2 -1
View File
@@ -68,7 +68,8 @@ def pil_to_temp_file(self, img: Image, dir: str, format="png") -> str: # pylint:
with tempfile.NamedTemporaryFile(delete=False, suffix=".png", dir=dir) as tmp:
name = tmp.name
img.save(name, pnginfo=(metadata if use_metadata else None))
shared.log.debug(f'Saving temp: image="{name}"')
size = os.path.getsize(name)
shared.log.debug(f'Saving temp: image="{name}" resolution={img.width}x{img.height} size={size}')
params = ', '.join([f'{k}: {v}' for k, v in img.info.items()])
params = params[12:] if params.startswith('parameters: ') else params
with open(os.path.join(paths.data_path, "params.txt"), "w", encoding="utf8") as file: