Adetailer扩展中实现自动保存遮罩图像的技术方案

背景介绍

Adetailer作为Stable Diffusion WebUI的一个强大扩展，主要用于图像细节增强和自动修复。在实际应用中，用户经常需要获取处理过程中生成的遮罩(mask)图像，以便进行后续处理或分析。本文将详细介绍如何在Adetailer中实现自动保存遮罩图像的功能。

技术实现原理

Adetailer的核心处理流程中已经包含了生成预览图像的功能，我们可以通过扩展这一功能来实现遮罩图像的保存。主要技术点包括：

1. 遮罩图像获取：从Adetailer处理结果中提取生成的遮罩
2. 图像处理：对原始遮罩进行必要的后处理（如模糊处理）
3. 保存机制：利用现有的图像保存功能实现遮罩的持久化存储

具体实现步骤

代码修改方案

在Adetailer的脚本文件(!adetailer.py)中，我们可以通过以下修改实现遮罩保存功能：

1. 首先需要导入必要的图像处理模块：

from PIL import Image, ImageFilter

2. 在图像处理流程中添加遮罩保存逻辑：

# 获取模糊处理后的遮罩
blurred_mask = masks[0].convert("L").filter(ImageFilter.GaussianBlur(args.ad_mask_blur))

# 保存纯遮罩图像
self.save_image(
    p,
    blurred_mask,
    condition="save_mask",
    suffix="-ad-mask" + suffix(n, "-"),
)

# 保存带遮罩的合成图像
composite = pp.image.copy()
composite.putalpha(blurred_mask)
self.save_image(
    p,
    composite,
    condition="save_mask_composite",
    suffix="-ad-mask-composite" + suffix(n, "-"),
)

功能说明

1. 遮罩模糊处理：使用高斯模糊对遮罩边缘进行处理，确保后续应用时过渡自然
2. 两种保存格式：
   • 纯遮罩图像：灰度图像，白色表示需要处理的区域
   • 合成图像：原始图像与遮罩的叠加，便于直观查看效果
3. 命名规范：自动添加"-ad-mask"或"-ad-mask-composite"后缀，便于识别

使用注意事项

1. 重启生效：修改代码后需要完全关闭Stable Diffusion WebUI(包括后台进程)才能使更改生效
2. 设置启用：需要在WebUI设置中启用"保存灰度遮罩"或"保存遮罩合成图"选项
3. 性能考虑：当前实现只保存第一个遮罩，如需保存多个需要进一步修改代码
4. 存储空间：保存额外图像会增加存储需求，特别是处理大量图片时

扩展思考

对于更复杂的需求，可以考虑以下增强功能：

1. 多遮罩保存：修改代码循环处理所有生成的遮罩
2. 自定义路径：实现独立的遮罩保存路径设置
3. 格式选择：支持不同图像格式(PNG/JPG等)的保存
4. 压缩选项：提供遮罩图像的压缩质量设置

总结

通过在Adetailer中实现遮罩自动保存功能，用户可以更方便地获取中间处理结果，为后续的图像处理流程提供更多可能性。这一改进虽然代码量不大，但能显著提升工作流程的灵活性和效率。开发者可以根据实际需求选择实现基础功能或进一步扩展更复杂的保存选项。