From 7b9ee43dcb4d79d43bbd4e0d18761344cb44d86c Mon Sep 17 00:00:00 2001
From: yangjiahua <kawa@yangjiahua.local>
Date: Fri, 11 Jan 2019 09:42:15 +0800
Subject: [PATCH] open file by with to avoid do not close the file if IOError

---
 Code/1_data_prepare/1_1_cifar10_to_png.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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 9df1499..2fb58cb 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):
-- 
GitLab