Keras Maxpooling2d layer gives ValueError

后端 未结 7 1695
难免孤独
难免孤独 2021-02-07 00:10

I am trying to replicate VGG16 model in keras, the following is my code:

model = Sequential()
model.add(ZeroPadding2D((1,1),input_shape=(3,224,224)))
model.add(C         


        
7条回答
  •  生来不讨喜
    2021-02-07 00:43

    For keras with TensorFlow try following:

    model.add(ZeroPadding2D((1, 1), input_shape=(img_rows, img_cols, channel)))
    

提交回复
热议问题