DenseNet121模型仅预测一类

Question

我想在类0和类1之间建立图像分类模型。
我从keras.applications导入使用DenseNet121模型

model = DenseNet121(weights= "imagenet", include_top=False, input_shape= (64,64,3),classes = 2)
x = model.output
x = GlobalAveragePooling2D()(x)
predictions = Dense(2, activation= 'softmax')(x)
model_2 = Model(inputs = model.input, outputs = predictions)
model_2.compile(optimizer= 'adam', loss='categorical_crossentropy', metrics=['accuracy'])
model_2.fit(
  X_train,
  to_categorical(y_train),
  epochs=10,
  validation_split=0.2,batch_size = 32
)
y_pr = model_2.predict(X_test)
y_re = np.argmax(y_pr,axis=1)

预测只是一个类别。
我尝试更改“优化程序”，但结果相同。有可能修复它吗？
非常感谢您在此问题上所付出的时间和协助