add image quality arg

https://github.com/Sanster/lama-cleaner/issues/229
2023-03-20 21:23:49 +08:00
parent 5f4c62ac18
commit 1bb25bebe6
5 changed files with 67 additions and 42 deletions
--- a/lama_cleaner/helper.py
+++ b/lama_cleaner/helper.py
@@ -135,9 +135,9 @@ def numpy_to_bytes(image_numpy: np.ndarray, ext: str) -> bytes:
    return image_bytes


-def pil_to_bytes(pil_img, ext: str, exif=None) -> bytes:
+def pil_to_bytes(pil_img, ext: str, quality: int = 95, exif=None) -> bytes:
    with io.BytesIO() as output:
-        pil_img.save(output, format=ext, exif=exif, quality=95)
+        pil_img.save(output, format=ext, exif=exif, quality=quality)
        image_bytes = output.getvalue()
    return image_bytes