diff --git a/Code/1_data_prepare/1_1_cifar10_to_png.py b/Code/1_data_prepare/1_1_cifar10_to_png.py
index 9df14994ab093c6f3f7a46203882690234b16f2e..2fb58cba00c1fc5d26e1827373ab418c03793759 100644
--- a/Code/1_data_prepare/1_1_cifar10_to_png.py
+++ b/Code/1_data_prepare/1_1_cifar10_to_png.py
@@ -16,9 +16,8 @@ Train = False   # 不解压训练集，仅解压测试集
 
 # 解压缩，返回解压后的字典
 def unpickle(file):
-    fo = open(file, 'rb')
-    dict_ = pickle.load(fo, encoding='bytes')
-    fo.close()
+    with open(file, 'rb') as fo:
+        dict_ = pickle.load(fo, encoding='bytes')
     return dict_
 
 def my_mkdir(my_dir):