FreeU

节点功能:FreeU 节点为 UNet 模型打上输出模块补丁,引入一个名为 FreeU(Free lunch for U-Net) 的机制,通过特定层级的特征通道缩放与频域滤波,增强图像纹理表现和清晰度,尤其是在高清、写实或结构图任务中效果显著。你可以把它理解为:“在不更换模型权重的前提下,让 UNet 输出得更清晰、更具细节”。

输入参数

参数名称说明
model输入的 UNet 模型对象。

输出参数

参数名称说明
MODEL应用 FreeU 策略后的 UNet 模型。(FreeU 是一种优化 UNet 中特征处理路径的方法,目的是提高生成图像的质量和稳定性。)

控件参数

参数名称说明
b1第一层主干通道的亮度增强系数。
b2第二层主干通道的亮度增强系数。
s1第一层辅助分支(跳跃连接)的频域压缩系数。
s2第二层辅助分支的频域压缩系数。

下图是效果对比,参数较多大家可以自行测试调整。