fix https://github.com/Sanster/lama-cleaner/issues/265

2023-04-16 11:00:04 +08:00 · 2023-04-16 11:00:04 +08:00 · 981ae3dc31
commit 981ae3dc31
parent 0baea45210
1 changed files with 2 additions and 1 deletions
--- a/lama_cleaner/server.py
+++ b/lama_cleaner/server.py
@ -129,7 +129,8 @@ def save_image():
    elif image.shape[2] == 4:
        image = cv2.cvtColor(image, cv2.COLOR_RGBA2BGRA)

-    cv2.imwrite(os.path.join(output_dir, filename), image)
+    save_path = os.path.join(output_dir, filename)
+    cv2.imencode(Path(save_path).suffix, image)[1].tofile(save_path)

    return "ok", 200