From f52db91713f0538a86de4ef9ce1eb4ddfa173867 Mon Sep 17 00:00:00 2001
From: yp <yp@bowong.ai>
Date: Wed, 11 Jun 2025 17:18:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BF=9D=E5=AD=98=E5=9B=BE?=
 =?UTF-8?q?=E7=89=87=E8=8A=82=E7=82=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 nodes/image.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/nodes/image.py b/nodes/image.py
index 2a8c73f..f9d88ad 100644
--- a/nodes/image.py
+++ b/nodes/image.py
@@ -21,7 +21,16 @@ class SaveImagePath:
     CATEGORY = "不忘科技-自定义节点🚩"
 
     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()
-        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)
+        # 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,)