let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add image quality arg

Browse Source 
https://github.com/Sanster/lama-cleaner/issues/229
This commit is contained in:
Qing
2023-03-20 21:23:49 +08:00
parent 5f4c62ac18
commit 1bb25bebe6
5 changed files with 67 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
lama_cleaner/server.py
View File

@@ -93,6 +93,7 @@ is_controlnet: bool = False
is_enable_file_manager: bool = False
is_enable_auto_saving: bool = False
is_desktop: bool = False
image_quality: int = 95
def get_image_ext(img_bytes):
@@ -300,10 +301,12 @@ def process():
ext = get_image_ext(origin_image_bytes)
# fmt: off
if exif is not None:
bytes_io = io.BytesIO(pil_to_bytes(Image.fromarray(res_np_img), ext, exif=exif))
bytes_io = io.BytesIO(pil_to_bytes(Image.fromarray(res_np_img), ext, quality=image_quality, exif=exif))
else:
bytes_io = io.BytesIO(pil_to_bytes(Image.fromarray(res_np_img), ext))
bytes_io = io.BytesIO(pil_to_bytes(Image.fromarray(res_np_img), ext, quality=image_quality))
# fmt: on
response = make_response(
send_file(
@@ -437,6 +440,10 @@ def main(args):
global output_dir
global is_enable_auto_saving
global is_controlnet
global image_quality
image_quality = args.quality
if args.sd_controlnet and args.model in SD15_MODELS:
is_controlnet = True