kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

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

Browse Source
This commit is contained in:
Qing 2023-04-16 11:00:04 +08:00
parent 0baea45210
commit 981ae3dc31
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lama_cleaner/server.py
View File

@ -129,7 +129,8 @@ def save_image():
elif image.shape[2] == 4: elif image.shape[2] == 4:
image = cv2.cvtColor(image, cv2.COLOR_RGBA2BGRA) 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 return "ok", 200