From 981ae3dc31ff20b5e425dcfc52330784d9e01d33 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sun, 16 Apr 2023 11:00:04 +0800
Subject: [PATCH] fix https://github.com/Sanster/lama-cleaner/issues/265

---
 lama_cleaner/server.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/lama_cleaner/server.py b/lama_cleaner/server.py
index b18f44a..5923da7 100644
--- 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