bowong
/
ComfyUI-CustomNode
6
0
Fork 0
Code Pull Requests Releases Activity

修改保存图片节点

This commit is contained in:
杨平 2025-06-11 17:18:38 +08:00
parent ddff65dff7
commit f52db91713
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/image.py
View File

@ -21,7 +21,16 @@ class SaveImagePath:
CATEGORY = "不忘科技-自定义节点🚩" CATEGORY = "不忘科技-自定义节点🚩"
def load(self, image_path:torch.Tensor): def load(self, image_path:torch.Tensor):
image_path = image_path.float()
# 假设数据范围在 [0, 255]，进行归一化
image_path = image_path / 255.0
# 检查并调整形状（这里只是示例，具体调整需根据实际情况）
if len(image_path.shape) == 3:
image_path = image_path.unsqueeze(0) # 添加批次维度
u = uuid.uuid4() u = uuid.uuid4()
p = os.path.join(os.path.dirname(os.path.abspath(__file__)), "output", "%s.jpg" % str(u)) p = os.path.join(os.path.dirname(os.path.abspath(__file__)), "output", "%s.jpg" % str(u))
torchvision.utils.save_image(image_path, p) torchvision.utils.save_image(image_path, p)
# u = uuid.uuid4()
# p = os.path.join(os.path.dirname(os.path.abspath(__file__)),"output","%s.jpg" % str(u))
# torchvision.utils.save_image(image_path, p)
return (p,) return (p,)