diff --git a/image_io.py b/image_io.py
index 43cdf02942dba8edf20d8cf391cc754c5b4be811..b6e73fa8b13c933eac591b7b1539c3cf1d67fa06 100644
--- a/image_io.py
+++ b/image_io.py
@@ -1,4 +1,4 @@
-# 导入必要的模块
+
 from image_utils import logging
 from skimage import io  # skimage.io模块用于读写图像文件
 
@@ -12,9 +12,12 @@ def load_image(image_path, as_gray=False):
     try:
         logging.info(f"开始载入图像: {image_path}")  # 记录开始载入图像的日志
         
-        # TODO: 载入图像文件的数据，如果as_gray为True，则载入灰度图像
-        # 注意: 如果as_gray为True 的时候 io.imread函数会将图像载入为类型为float64的灰度图像，范围是[0,1]，需要转换为uint8类型，范围是[0,255]
+        # 载入图像文件的数据，如果as_gray为True，则载入灰度图像
+        image = io.imread(image_path, as_gray=as_gray)
         
+        # 如果as_gray为True 的时候 io.imread函数会将图像载入为类型为float64的灰度图像，范围是[0,1]，需要转换为uint8类型，范围是[0,255]
+        if as_gray and image.dtype == np.float64:
+            image = (image * 255).astype(np.uint8)
         
         logging.info(f"成功载入图像: {image_path}")  # 记录成功载入图像的日志
         return image
@@ -32,8 +35,8 @@ def save_image(image, save_path, format='png'):
     try:
         logging.info(f"开始保存图像: {save_path}")  # 记录开始保存图像的日志
         
-        # TODO: 将图像 image 保存到文件 save_path
-        
+        # 将图像 image 保存到文件 save_path
+        io.imsave(save_path, image, format=format)
         
         logging.info(f"图像成功保存在: {save_path}")  # 记录成功保存图像的日志
     except Exception as e: