ValueError: Unexpectedly found an instance of type `<class 'keras.lay…

…ers.convolutional.ZeroPadding2D'>`. Expected a symbolic tensor instance.
cl0ver012 · May 13, 2018 · c6f3eb1 · c6f3eb1
1 parent bab5f66
commit c6f3eb1
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/model.py b/model.py
@@ -1,12 +1,13 @@
 import keras.backend as K
-from keras.layers import Conv2D, UpSampling2D, BatchNormalization, ZeroPadding2D, MaxPooling2D
+from keras.layers import Input, Conv2D, UpSampling2D, BatchNormalization, ZeroPadding2D, MaxPooling2D
 from keras.models import Model
 
 
 def create_model(img_rows, img_cols, channel=4):
     # Encoder
-    input_tensor = ZeroPadding2D((1, 1), input_shape=(img_rows, img_cols, channel), name='input')
-    x = Conv2D(64, (3, 3), activation='relu', name='conv1_1')(input_tensor)
+    input_tensor = Input(shape=(img_rows, img_cols, channel), name='input')
+    x = ZeroPadding2D((1, 1))(input_tensor)
+    x = Conv2D(64, (3, 3), activation='relu', name='conv1_1')(x)
     x = ZeroPadding2D((1, 1))(x)
     x = Conv2D(64, (3, 3), activation='relu', name='conv1_2')(x)
     x = MaxPooling2D((2, 2), strides=(2, 2))(x)