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>
76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
import torch
|
|
import numpy as np
|
|
from PIL import Image
|
|
import os
|
|
|
|
class LoadImageByPathNode:
|
|
"""
|
|
按路径加载图像的节点
|
|
"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"image_path": ("STRING", {
|
|
"default": "E:\\ComfyUIModels\\input\\10\\1.png",
|
|
"multiline": False,
|
|
"tooltip": "图像文件的完整路径，可指向 ComfyUI 目录之外的任意位置。\n"
|
|
"适合批量处理散落在各处的素材，省去先拷进 input 目录。\n"
|
|
"Windows 路径直接粘贴即可，反斜杠无需转义。"
|
|
}),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
FUNCTION = "load_image_by_path"
|
|
CATEGORY = "Rui-Node🐶/文件存储与加载📁"
|
|
|
|
def load_image_by_path(self, image_path):
|
|
"""
|
|
根据指定路径加载图像
|
|
|
|
参数:
|
|
image_path: 图像文件的完整路径
|
|
|
|
返回:
|
|
加载的图像张量
|
|
"""
|
|
try:
|
|
# 检查文件是否存在
|
|
if not os.path.exists(image_path):
|
|
raise FileNotFoundError(f"图像文件不存在: {image_path}")
|
|
|
|
# 使用 PIL 加载图像
|
|
img_pil = Image.open(image_path)
|
|
|
|
# 确保图像为 RGB 模式
|
|
if img_pil.mode != 'RGB':
|
|
img_pil = img_pil.convert('RGB')
|
|
|
|
# 将 PIL 图像转换为 NumPy 数组
|
|
img_np = np.array(img_pil).astype(np.float32) / 255.0
|
|
|
|
# 转换为 PyTorch 张量并添加批次维度
|
|
# ComfyUI 期望的格式为 BHWC
|
|
img_tensor = torch.from_numpy(img_np).unsqueeze(0)
|
|
|
|
return (img_tensor,)
|
|
|
|
except Exception as e:
|
|
# 如果加载失败，创建一个默认的黑色图像
|
|
print(f"加载图像失败: {str(e)}")
|
|
# 创建一个 512x512 的黑色图像作为默认值
|
|
default_img = np.zeros((512, 512, 3), dtype=np.float32)
|
|
default_tensor = torch.from_numpy(default_img).unsqueeze(0)
|
|
return (default_tensor,)
|
|
|
|
# 节点映射字典，用于 ComfyUI 注册节点
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LoadImageByPath": LoadImageByPathNode
|
|
}
|
|
|
|
# 节点显示名称映射，用于在 UI 中显示友好名称
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LoadImageByPath": "按路径加载图像 / Load Image By Path"
|
|
} |