前提
GoogleColabで画像分類を試しています。
YouTubeの動画を参考にしながら進めていたところ、エラーで躓きました。
参考YouTube→https://youtu.be/i3ZIRJYlzkU
精度の確認を実装中に以下のエラーメッセージが発生しました。
ソースコード①学習データとテストデータに分ける
#テストデータが2割、学習データが8割に自動で分ける X_train, X_test, y_train, y_test=train_test_split(X,Y,test_size=0.2) print(X_train.shape) print(X_test.shape) print(y_train.shape) print(y_test.shape)
ソースコード②学習モデルの準備
from keras.applications.vgg16 import VGG16 from keras.models import Sequential from keras.models import model_from_json from keras.models import Model from keras.layers import Input,Activation,merge,Dense,Flatten,Dropout from tensorflow.keras.optimizers import Adam #vgg16をベースモデルとして読み込む input_tensor=Input(shape=(244,244,3)) base_model=VGG16(weights='imagenet',input_tensor=input_tensor,include_top=False) #後付け→トップモデルを2クラスに変更する top_model=Sequential() top_model.add(Flatten(input_shape=base_model.output_shape[1:])) top_model.add(Dense(n_classes,activation='softmax')) #がっちゃんこ→新しい学習モデル model=Model(inputs=base_model.input,outputs=top_model(base_model.output)) #学習させない層→頭から15層までは学習しなくていい for layer in model.layers[:15]: layer.trainable=False print('#layers=',len(model.layers)) model.compile(loss='categorical_crossentropy', optimizer='adam',metrics=['accuracy']) model.summary() #layers= 20 20層 #Total params: 14,764,866 #Trainable params: 7,129,602→ #Non-trainable params: 7,635,264
発生している問題・エラーメッセージ
Epoch 1/20 --------------------------------------------------------------------------- ValueError Traceback (most recent call last) <ipython-input-14-f1ea30a5d471> in <module> 1 #trainデータで学習 ----> 2 model.fit (X_train,y_train,epochs=20, batch_size=16) 3 #テストデータで精度確認 4 score=model.evaluate(X_test,y_test,batch_size=16) 1 frames /usr/local/lib/python3.7/dist-packages/tensorflow/python/framework/func_graph.py in autograph_handler(*args, **kwargs) 1145 except Exception as e: # pylint:disable=broad-except 1146 if hasattr(e, "ag_error_metadata"): -> 1147 raise e.ag_error_metadata.to_exception(e) 1148 else: 1149 raise ValueError: in user code: File "/usr/local/lib/python3.7/dist-packages/keras/engine/training.py", line 1021, in train_function * return step_function(self, iterator) File "/usr/local/lib/python3.7/dist-packages/keras/engine/training.py", line 1010, in step_function ** outputs = model.distribute_strategy.run(run_step, args=(data,)) File "/usr/local/lib/python3.7/dist-packages/keras/engine/training.py", line 1000, in run_step ** outputs = model.train_step(data) File "/usr/local/lib/python3.7/dist-packages/keras/engine/training.py", line 859, in train_step y_pred = self(x, training=True) File "/usr/local/lib/python3.7/dist-packages/keras/utils/traceback_utils.py", line 67, in error_handler raise e.with_traceback(filtered_tb) from None File "/usr/local/lib/python3.7/dist-packages/keras/engine/input_spec.py", line 264, in assert_input_compatibility raise ValueError(f'Input {input_index} of layer "{layer_name}" is ' ValueError: Input 0 of layer "model" is incompatible with the layer: expected shape=(None, 244, 244, 3), found shape=(None, 224, 224, 3)
該当のソースコード
#trainデータで学習 model.fit (X_train,y_train,epochs=20, batch_size=16) #テストデータで精度確認 score=model.evaluate(X_test,y_test,batch_size=16)
試したこと
別のノートで再度実行→エラー
回答1件
あなたの回答
tips
プレビュー