diff --git a/py/image_shift.py b/py/image_shift.py index 17bd7c6..4fd78e5 100644 --- a/py/image_shift.py +++ b/py/image_shift.py @@ -15,7 +15,7 @@ class ImageShift: "image": ("IMAGE", ), # "shift_x": ("INT", {"default": 256, "min": -9999, "max": 9999, "step": 1}), "shift_y": ("INT", {"default": 256, "min": -9999, "max": 9999, "step": 1}), - "cyclic": ("BOOLEAN", {"default": True}), # 反转mask# + "cyclic": ("BOOLEAN", {"default": True}), # 是否循环重复 "background_color": ("STRING", {"default": "#000000"}), "border_mask_width": ("INT", {"default": 20, "min": 0, "max": 999, "step": 1}), "border_mask_blur": ("INT", {"default": 12, "min": 0, "max": 999, "step": 1}), diff --git a/py/lama.py b/py/lama.py index eb2c05f..de2d678 100644 --- a/py/lama.py +++ b/py/lama.py @@ -1,9 +1,8 @@ import os.path import shutil from pathlib import Path - from .imagefunc import * -from iopaint import cli + NODE_NAME = 'LaMa' @@ -93,6 +92,7 @@ class LaMa: log(f"Error: {NODE_NAME} skipped, because unable to create temporary file.", message_type='error') return (image, ) # process + from iopaint import cli cli.run(model=lama_model, device=device, image=Path(image_dir), mask=Path(mask_dir), output=Path(result_dir)) ret_image = check_image_file(os.path.join(result_dir, file_name), 500) shutil.rmtree(temp_dir)