comfyui-inpainting

ComfyUI 局部重绘 (Inpainting)

修改图像的特定区域，保持其他部分不变。

---

预注入变量

run_python 已注入：submit_workflow, save_preview

---

核心概念

局部重绘需要：

1. 原图 - 要修改的图像
2. 遮罩 (Mask) - 标记要修改的区域（白色=修改，黑色=保持）
3. 提示词 - 描述修改后的内容

---

使用方法

方法 1: 使用便捷函数

from comfyui_inpainting import build_inpaint_workflow

wf = build_inpaint_workflow(
    image_path="original.png",
    mask_path="mask.png",  # 白色区域会被重绘
    prompt="beautiful red dress",  # 修改后的内容
    negative_prompt="low quality",
    checkpoint="sdxl_base.safetensors",
    denoise=1.0,  # 1.0 = 完全重绘遮罩区域
    seed=42
)

result = submit_workflow(wf)
if result.get("images"):
    save_preview(result["images"][0])

方法 2: 分步构建

import random

wf = {}

# 1. 加载模型
wf["1"] = {"class_type": "CheckpointLoaderSimple", "inputs": {"ckpt_name": "sdxl.safetensors"}}

# 2. 加载原图和遮罩
wf["2"] = {"class_type": "LoadImage", "inputs": {"image": "original.png"}}
wf["3"] = {"class_type": "LoadImageMask", "inputs": {"image": "mask.png", "channel": "alpha"}}

# 3. 编码提示词
wf["4"] = {"class_type": "CLIPTextEncode", "inputs": {"text": "beautiful red dress", "clip": ["1", 1]}}
wf["5"] = {"class_type": "CLIPTextEncode", "inputs": {"text": "low quality", "clip": ["1", 1]}}

# 4. VAE 编码原图
wf["6"] = {"class_type": "VAEEncode", "inputs": {"pixels": ["2", 0], "vae": ["1", 2]}}

# 5. 设置潜空间噪波（只在遮罩区域）
wf["7"] = {
    "class_type": "SetLatentNoiseMask",
    "inputs": {
        "samples": ["6", 0],
        "mask": ["3", 0]
    }
}

# 6. 采样（只在遮罩区域重绘）
wf["8"] = {
    "class_type": "KSampler",
    "inputs": {
        "model": ["1", 0],
        "positive": ["4", 0],
        "negative": ["5", 0],
        "latent_image": ["7", 0],
        "seed": random.randint(0, 2**32),
        "steps": 25,
        "cfg": 7.0,
        "sampler_name": "euler",
        "scheduler": "normal",
        "denoise": 1.0  # 只在遮罩区域生效
    }
}

# 7. 解码保存
wf["9"] = {"class_type": "VAEDecode", "inputs": {"samples": ["8", 0], "vae": ["1", 2]}}
wf["10"] = {"class_type": "SaveImage", "inputs": {"images": ["9", 0], "filename_prefix": "inpaint"}}

result = submit_workflow(wf)