「VGG16の出力のカテゴリの数を3つに変更するとエラーが出る。」
下記のサイトと同じことをしているのですが、キャラクターの種類を5人→3人に変更しようとしています。
VGG16を転移学習させて「まどか☆マギカ」のキャラを見分ける
キャラクターの画像の種類を3つに変更し(train, test, validation全て)、
n_categories=3
に変え、「ライブラリのインポートとモデル構築」の項まではできたのですが、「画像の前処理をして学習」の項のプログラムを実行しようとすると
ValueError: Error when checking target: expected dense_8 to have shape (3,) but got array with shape (4,)
というエラーが出てしまいます。n_categories=4なら通ったのですが、なぜn_categories=3だと上記の様なエラーが出てしまうのでしょうか?
なぜn_categories=3だと上記の様なエラーが出てしまうかわかる方はいますか?
一応、筆者には連絡していますが、返信がないので、ここに投稿させていただきました。
実行環境は以下の通りです。
python 3.7
jupiter notebook 6.0.3
anaconda3
windows10 64bit
あなたの回答
tips
プレビュー