【发布时间】:2018-10-09 10:00:06
【问题描述】:
我的模型定义如下:
def build(data):
model = Sequential()
model.add(Cropping2D(cropping=((79, 145), (50, 250)), input_shape=
(160,320,3)))
model.add(Lambda(lambda x: x/127.5 - 1.0))
model.add(Conv2D(24, (2, 2), padding='same'))
model.add(ELU())
model.add(Conv2D(36, (2, 2), padding='same'))
model.add(ELU())
model.add(Conv2D(48, (2, 2), padding='same'))
model.add(ELU())
# Add a flatten layer
model.add(Flatten())
model.summary()
model.add(Dense(100))
model.add(ELU())
model.add(Dense(50))
model.add(ELU())
model.add(Dense(10))
model.add(ELU())
model.add(Dense(1))
return model
得到这个错误:
ValueError:
Dense的输入的最后一个维度应该是 定义。找到None。
我运行model.summary() 并得到以下输出
Layer (type) Output Shape Param #
=================================================================
cropping2d_15 (Cropping2D) (None, 0, 20, 3) 0
_________________________________________________________________
lambda_23 (Lambda) (None, 0, 20, 3) 0
_________________________________________________________________
conv2d_47 (Conv2D) (None, 0, 20, 24) 312
_________________________________________________________________
elu_43 (ELU) (None, 0, 20, 24) 0
_________________________________________________________________
conv2d_48 (Conv2D) (None, 0, 20, 36) 3492
_________________________________________________________________
elu_44 (ELU) (None, 0, 20, 36) 0
_________________________________________________________________
conv2d_49 (Conv2D) (None, 0, 20, 48) 6960
_________________________________________________________________
elu_45 (ELU) (None, 0, 20, 48) 0
_________________________________________________________________
flatten_12 (Flatten) (None, None) 0
=================================================================
Total params: 10,764
Trainable params: 10,764
Non-trainable params: 0
我对 python 还很陌生,任何输入将不胜感激。
【问题讨论】:
-
我的问题是为什么 Flatten 层会给出 (None, None) 的输出。
标签: python tensorflow keras deep-learning conv-neural-network