kerasのVGG16モデルに対し、グレースケールの画像を入力する方法（ImageDataGenerator）

表記の件について、次のサイトを参考にプログラムを作成したところ、callbacksに対し下記のようなエラーが発生しました。サイトの通り、RGBの(256,256,3)の時は問題なく動いたのですが、グレースケールにしたらエラーが出るようになりました。どうかアドバイスをお願いしたいです。よろしくお願いします。
また、RGBとグレースケールの比較をしたいので、グレースケール画像をRGB画像に変換するという方法で解決する予定はありません。keras.jsonはchannel_lastに設定しています。

・参考にしたサイト
VGG16を転移学習させて「まどか☆マギカ」のキャラを見分ける
 kerasのVGG16モデルに対し、グレースケールの画像を入力する方法について

・エラー文

ValueError: Error when checking input: expected input_17 to have shape (198, 386, 1) but got array with shape (198, 386, 3)

行動規範の内容に同意します

回答1件

ベストアンサー

color_mode= を追加したら、エラー出ませんでした

python
1train_generator=train_datagen.flow_from_directory(
2train_dir,
3target_size=(img_width,img_height),
4batch_size=batch_size,
5classes=classes,
6class_mode='categorical',
7shuffle=True,
8seed=seed,
9color_mode='grayscale'
10)