Merge pull request #17 from MuLx10/patch

Fixes the issue(#16)
imlab-uiip · Mar 6, 2018 · 0304b85 · 0304b85
2 parents a7e50c9 + 10c11d4
commit 0304b85
Show file tree

Hide file tree

Showing 2 changed files with 1,611 additions and 590 deletions.
diff --git a/build_model.py b/build_model.py
@@ -11,10 +11,10 @@
 kernel = 3
 
 encoding_layers = [
-    Convolution2D(64, kernel, kernel, border_mode='same', input_shape=(1, img_h, img_w)),
+    Convolution2D(64, kernel, border_mode='same', input_shape=( img_h, img_w,1)),
     BatchNormalization(),
     Activation('relu'),
-    Convolution2D(64, kernel, kernel, border_mode='same'),
+    Convolution2D(64, kernel, border_mode='same'),
     BatchNormalization(),
     Activation('relu'),
     MaxPooling2D(),
@@ -66,6 +66,7 @@
 
 for l in autoencoder.encoding_layers:
     autoencoder.add(l)
+    print(l.input_shape,l.output_shape,l)
 
 decoding_layers = [
     UpSampling2D(),