增加npy文件转成图片的脚本

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)

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