rui40000andClaude Opus 4.8
5d3834503e
【修复】角色被格线切断（脚、手杖、飘起的斗篷被削掉） 新增 expand_beyond_cell（默认开启）：格子只用来判定「这是哪个方向」， 角色的实际范围由它自身的连通区域决定，按质心归属确保邻居不混入。 实测 8/8 方向的裁剪框边缘 alpha 从 1.00（内容顶到边界=被切断） 降到 0.00，S 方向高度 326→356、E 方向宽度 150→188 把缺的部分找了回来。 代价是需要两遍扫描（先求全序列并集框再提取），耗时 4.7s→14.9s。 【规则】每个参数都必须有中文 tooltip，作为以后的统一约定 全仓库 26 个节点 169 个参数，此前缺 115 个，现已 100% 覆盖。 tooltip 写「怎么调」而不只是「是什么」：给取值区间的实际影响、 推荐值与踩坑提示（如 OpenAI/ZenMux 的地址栏不能带 :// ， 素材拆分节点用于动画序列时顺序会漂移等）。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import torch
|
|
import numpy as np
|
|
from PIL import Image, ImageEnhance
|
|
|
|
class SaturationNode:
|
|
"""
|
|
调整图像饱和度的节点
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image": ("IMAGE", {
|
|
"tooltip": "要调整饱和度的图像。"
|
|
}),
|
|
"saturation": ("FLOAT", {
|
|
"default": 1.0,
|
|
"min": 0.0,
|
|
"max": 5.0,
|
|
"step": 0.1,
|
|
"tooltip": "饱和度系数：\n"
|
|
"0.0 = 完全去色（黑白）\n"
|
|
"1.0 = 保持原样\n"
|
|
"1.0~2.0 = 色彩逐渐浓郁，超过 2.0 容易溢出失真"
|
|
}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
FUNCTION = "adjust_saturation"
|
|
CATEGORY = "Rui-Node🐶/图像调节🎨"
|
|
|
|
def adjust_saturation(self, image, saturation):
|
|
"""
|
|
调整输入图像的饱和度
|
|
|
|
参数:
|
|
image: 输入图像张量 (B, H, W, C) 格式
|
|
saturation: 饱和度调整系数，1.0为原始饱和度
|
|
|
|
返回:
|
|
调整后的图像张量
|
|
"""
|
|
# 将图像从 PyTorch 张量转换为 PIL 图像进行处理
|
|
batch_size = image.shape[0]
|
|
result = []
|
|
|
|
for i in range(batch_size):
|
|
# 将单个图像从 PyTorch 张量转换为 NumPy 数组
|
|
# ComfyUI 中图像格式为 BHWC，值范围为 0-1
|
|
img_np = image[i].cpu().numpy()
|
|
|
|
# 确保值范围在 0-1 之间
|
|
img_np = np.clip(img_np, 0, 1)
|
|
|
|
# 转换为 PIL 图像 (值范围 0-255)
|
|
img_pil = Image.fromarray((img_np * 255).astype(np.uint8), 'RGB')
|
|
|
|
# 使用 PIL 的 ImageEnhance.Color 调整饱和度
|
|
enhancer = ImageEnhance.Color(img_pil)
|
|
enhanced_img = enhancer.enhance(saturation)
|
|
|
|
# 将处理后的图像转回 NumPy 数组，然后转为 PyTorch 张量
|
|
enhanced_np = np.array(enhanced_img).astype(np.float32) / 255.0
|
|
enhanced_tensor = torch.from_numpy(enhanced_np)
|
|
|
|
result.append(enhanced_tensor)
|
|
|
|
# 将结果堆叠为批次
|
|
return (torch.stack(result),)
|
|
|
|
# 节点映射字典，用于 ComfyUI 注册节点
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SaturationAdjustment": SaturationNode
|
|
}
|
|
|
|
# 节点显示名称映射，用于在 UI 中显示友好名称
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SaturationAdjustment": "调整饱和度 / Saturation Adjustment"
|
|
} |