From 2da270236ee9fb7ca31dd2b3a7ac5c26d5e0814c Mon Sep 17 00:00:00 2001
From: sjh00 <sjh00@outlook.com>
Date: Tue, 3 Jun 2025 21:31:28 +0800
Subject: [PATCH] fix

---
 load_image_with_info.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/load_image_with_info.py b/load_image_with_info.py
index 0fd80b1..aa950e1 100644
--- a/load_image_with_info.py
+++ b/load_image_with_info.py
@@ -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