Keras 尝试拟合类似VGG的模型时出现类型错误

当我尝试适合以下模型时：

model = Sequential([
    Lambda(vgg_preprocess, input_shape=(3,244,244)),
    Conv2D(64,3,3, activation='relu'),
    BatchNormalization(axis=1),
    Conv2D(64,3,3, activation='relu'),
    MaxPooling2D(),
    BatchNormalization(axis=1),
    Conv2D(128,3,3, activation='relu'),
    BatchNormalization(axis=1),
    Conv2D(128,3,3, activation='relu'),
    MaxPooling2D(),
    BatchNormalization(axis=1),
    Conv2D(256,3,3, activation='relu'),
    BatchNormalization(axis=1),
    Conv2D(256,3,3, activation='relu'),
    MaxPooling2D(),
    Flatten(),
    BatchNormalization(),
    Dense(1024, activation='relu'),
    BatchNormalization(),
    Dropout(0.5),
    Dense(1024, activation='relu'),
    BatchNormalization(),
    Dense(10, activation='softmax')
])
model.compile(Adam(), loss='categorical_crossentropy', metrics=['accuracy'])

我得到这个错误：

TypeError: Cannot convert Type TensorType(float32, 4D) (of Variable AbstractConv2d_gradInputs{convdim=2, border_mode='valid', subsample=(1, 1), filter_flip=True, imshp=(None, 256, 56, 56), kshp=(256, 256, 3, 3), filter_dilation=(1, 1)}.0) into Type TensorType(float64, 4D). You can try to manually convert AbstractConv2d_gradInputs{convdim=2, border_mode='valid', subsample=(1, 1), filter_flip=True, imshp=(None, 256, 56, 56), kshp=(256, 256, 3, 3), filter_dilation=(1, 1)}.0 into a TensorType(float64, 4D).

我是这样做的：

 model.fit_generator(train_batches, train_batches.n, nb_epoch=1, validation_data=test_batches, nb_val_samples=test_batches.n)

下面是

vgg_预处理

函数：

vgg_mean = np.array([123.68, 116.779, 103.939]).reshape((3,1,1))

def vgg_preprocess(x):
    x = x - vgg_mean  #Subtract the mean of each channel
    return x[:, ::-1] #Inverse the channel order to suit that of VGG RGB->BGR

---

这意味着什么，如何修复它？

问题在于

vgg_mean.dtype='float64

，而大多数DL包中的标准浮点决定是

float32

设置：

vgg_mean = np.array(vgg_mean, dtype='float32')

---

应该可以解决您的问题。

vgg_预处理的样子如何？@MarcinMożejko它可以规范化输入并重新排列颜色通道。我把它添加到了问题中。你能打印出

vgg\u-mean.dtype

吗？试试

vgg\u-mean=np.array（vgg\u-mean，dtype='float32'）

。10分钟后，好吗？