文字認識モデルについて

Question

文系初心者です。
文字認識モデルを作成しようと思いましたが、
下記のエラーが発生しました。
何卒よろしくお願い致します。

ValueError: Error when checking input: expected conv2d_11_input to have 4 dimensions, but got array with shape (2100, 28, 28)

該当のソースコード

log = model.fit(train,train_labels,
                batch_size = 256,
                epochs = 16,
                validation_data = (test,test_labels))

ソースコード

import keras
from keras.models import Sequential
from keras.layers import Conv2D,MaxPooling2D,Activation,Flatten,Dense
from keras.optimizers import Adadelta
import numpy as np
import matplotlib.pyplot as plt

model = Sequential()

model.add(Conv2D(3,(2,2),input_shape = (28,28,1)))
model.add(Activation("relu"))

model.add(Conv2D(6,(2,2)))
model.add(Activation("relu"))

model.add(MaxPooling2D(pool_size = (2,2)))

model.add(Conv2D(9,(2,2)))
model.add(Activation("relu"))

model.add(Conv2D(12,(2,2)))
model.add(Activation("relu"))

model.add(MaxPooling2D(pool_size = (2,2)))

model.add(Flatten())

model.add(Dense(1024))
model.add(Activation("relu"))

model.add(Dense(512))
model.add(Activation("relu"))

model.add(Dense(256))
model.add(Activation("relu"))

model.add(Dense(10))
model.add(Activation("softmax"))

model.compile(loss = keras.losses.categorical_crossentropy,
              optimizer = Adadelta(),
              metrics = ["accuracy"])



train_data = np.load("train_data.npy")
train_label = np.load("train_label.npy")

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(train_data, train_label, 
                                                                                        test_size=0.3, random_state=1234,
                                                                                        shuffle=True
                                                                                       )


from sklearn.preprocessing import LabelBinarizer
lb = LabelBinarizer()

train = X_train/255
test = X_test/255
train = train.reshape(-1, 28*28)
test = test.reshape(-1, 28*28)
train_labels = lb.fit_transform(y_train)
test_labels = lb.fit_transform(y_test)

train = train.reshape(-1, 1, 28, 28)
test = test.reshape(-1, 1, 28, 28)

train= train.reshape(-1, 28, 28)



log = model.fit(train,train_labels,
                batch_size = 256,
                epochs = 16,
                validation_data = (test,test_labels))

入力次元が違っているのかと思っているのですがどうでしょうか？
画像はカラーの手書き文字です。
初心者のため不備等あるかと思いますが何卒よろしくお願い致します。

Answer 1

入力データとして

model.add(Conv2D(3,(2,2),input_shape = (28,28,1)))
model.add(Activation("relu"))

28x28x1 のデータを渡すように記述されておりますが、実際の入力として

train= train.reshape(-1, 28, 28)

と 28x28 のデータが渡っております。

train= train.reshape(-1, 28, 28, 1)

とする必要があるのではないでしょうか

Comments

[yusuke_yamamoto]

回答ありがとうございます。
ご指摘された箇所を修正して改善されました。
初心者ですので本当に助かります。