fix

2025-06-03 21:31:28 +08:00
parent 6a4d2eab7e
commit 2da270236e
1 changed files with 1 additions and 1 deletions
@@ -189,7 +189,7 @@ class SaveImageWithInfo:
            exif_data = {}
        
        # 保存图像
-        img = 255. * image.cpu().numpy()
+        img = 255. * image[0].cpu().numpy()
        img = Image.fromarray(np.clip(img, 0, 255).astype(np.uint8))
        
        # 设置DPI