fcf/migan fix change original image

This commit is contained in:
Qing 2024-01-04 21:39:09 +08:00
parent 9df94489e1
commit cd82b21cd9
2 changed files with 2 additions and 2 deletions

View File

@ -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

View File

@ -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