From cd82b21cd9ddd76db32f244a53cdcda92e395c42 Mon Sep 17 00:00:00 2001 From: Qing Date: Thu, 4 Jan 2024 21:39:09 +0800 Subject: [PATCH] fcf/migan fix change original image --- lama_cleaner/model/fcf.py | 2 +- lama_cleaner/model/mi_gan.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lama_cleaner/model/fcf.py b/lama_cleaner/model/fcf.py index 10e9113..327b533 100644 --- a/lama_cleaner/model/fcf.py +++ b/lama_cleaner/model/fcf.py @@ -1698,7 +1698,7 @@ class FcF(InpaintModel): crop_result.append((inpaint_result, crop_box)) - inpaint_result = image[:, :, ::-1] + inpaint_result = image[:, :, ::-1].copy() for crop_image, crop_box in crop_result: x1, y1, x2, y2 = crop_box inpaint_result[y1:y2, x1:x2, :] = crop_image diff --git a/lama_cleaner/model/mi_gan.py b/lama_cleaner/model/mi_gan.py index bd85ee2..923e6ba 100644 --- a/lama_cleaner/model/mi_gan.py +++ b/lama_cleaner/model/mi_gan.py @@ -73,7 +73,7 @@ class MIGAN(InpaintModel): crop_result.append((inpaint_result, crop_box)) - inpaint_result = image[:, :, ::-1] + inpaint_result = image[:, :, ::-1].copy() for crop_image, crop_box in crop_result: x1, y1, x2, y2 = crop_box inpaint_result[y1:y2, x1:x2, :] = crop_image