tensorflowを用いてCNNにより2分類を行いたいのですが,
畳み込み層が3層までならコードは回るのですが,4層以上になるとエラーが出てしまいます。
モデルは下のコードのようになっております。
宜しくお願いします。```
model = tf.keras.models.Sequential()
model.add( tf.keras.layers.Reshape((28, 28, 1), input_shape=(28, 28)) )
model.add( tf.keras.layers.Conv2D(32, (3, 3), activation='relu') )
model.add( tf.keras.layers.MaxPooling2D(pool_size=(2,2)) )
model.add( tf.keras.layers.Conv2D(32, (3, 3), activation='relu') )
model.add( tf.keras.layers.MaxPooling2D(pool_size=(2,2)) )
model.add( tf.keras.layers.Conv2D(64, (3, 3), activation='relu') )
model.add( tf.keras.layers.MaxPooling2D(pool_size=(2,2)) )
model.add( tf.keras.layers.Conv2D(64, (3, 3), activation='relu') ) model.add( tf.keras.layers.MaxPooling2D(pool_size=(2,2)) ) model.add( tf.keras.layers.Flatten() ) model.add( tf.keras.layers.Dense(64, activation='relu') ) model.add( tf.keras.layers.Dropout(0.2) ) model.add( tf.keras.layers.Dense(2, activation='softmax') )
コード
あなたの回答
tips
プレビュー