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