neural-network - 输入形状错误用 resnet keras 训练一些数据

Question

我正在尝试用 CNN 训练一些数据。

   x_train.shape
    (67197, 99, 81, 1)
   y_train.shape
    (67197, 12)

并尝试使用 Keras 的 Resnet 方法

import keras
import keras_resnet.models

input_shape = (98,81,1)
nclass = 12

    x = keras.layers.Input(input_shape)
    model = keras_resnet.models.ResNet50(x,classes=nclass)
    model.compile("adam","categorical_crossentropy",["accuracy"])
    model.fit(x_train,y_train,
             batch_size = 300,
             nb_epoch=5,
             validation_data = (x_test,y_test),
             shuffle = True,
             )

但我得到了一些形状错误。

ValueError: Error when checking input: expected input_3 to have shape (None, 98, 81, 1) but got array with shape (67197, 99, 81, 1)

Answer 1

它只是一个错字，您的训练数据具有形状（样本，99,81,1），因此输入形状应该是（99,81,1），而不是（98,81,1）。它只差一个。