From 732383a42001274ba65048811a2a1f4a8eaf2e8b Mon Sep 17 00:00:00 2001
From: tomguluson92 <314913739@qq.com>
Date: Sat, 23 Sep 2017 07:30:44 +0800
Subject: [PATCH 1/2] Update Unet.py for directory find error

---
 unet.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/unet.py b/unet.py
index f83921c98..e1e6d19a1 100644
--- a/unet.py
+++ b/unet.py
@@ -1,3 +1,5 @@
+import sys
+sys.path.append('./')
 import numpy as np
 from keras.models import *
 from keras.layers import Input, merge, Conv2D, MaxPooling2D, UpSampling2D, Dropout, Cropping2D

From fa9478eae0d8a61455735087614595e95e78c8e0 Mon Sep 17 00:00:00 2001
From: tomguluson92 <314913739@qq.com>
Date: Sat, 23 Sep 2017 07:32:02 +0800
Subject: [PATCH 2/2] add a script for transfer .npy to imgs

---
 save_png.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 save_png.py

diff --git a/save_png.py b/save_png.py
new file mode 100644
index 000000000..69eb9b633
--- /dev/null
+++ b/save_png.py
@@ -0,0 +1,25 @@
+'''
+    tranfer the test dataset output to imgs
+'''
+# coding:UTF-8
+from keras.preprocessing.image import array_to_img
+import numpy as np
+
+
+def save_output(default_dir):
+    '''
+    save test dataset into imgs into a default dir
+    '''
+
+    imgs_train = np.load(file= "./imgs_mask_test.npy")
+    imgs_train = imgs_train.astype('float32')
+
+    pictures = np.shape(imgs_train)[0]
+
+    for i in range(pictures):
+        img_tmp = array_to_img(imgs_train[i])
+        img_tmp.save(default_dir+"{}.".format(i)+"png")
+
+if __name__ == "__main__":
+    default_dir = "./aoao/"
+    save_output(default_dir)
\ No newline at end of file