我目前正在做一个项目,需要在一组图像中预测眼部疾病。我正在使用 Keras 内置应用程序。我在 VGG16 和 VGG19 上得到了很好的结果,但在 Xception 架构上,我每个时期的 AUC 都恰好为 0.5。
我尝试了不同的优化器和学习率,但没有任何效果。我通过从 RMSProp 优化器切换到 Adam 优化器解决了 VGG19 的相同问题,但我无法让它适用于 Xception。
def buildModel():
from keras.models import Model
from keras.layers import Dense, Flatten
from keras.optimizers import adam
input_model = applications.xception.Xception(
include_top=False,
weights='imagenet',
input_tensor=None,
input_shape=input_sizes["xception"],
pooling=None,
classes=2)
base_model = input_model
x = base_model.output
x = Flatten()(x)
predictions = Dense(2, activation='softmax')(x)
model = Model(inputs=base_model.input, outputs=predictions)
for layer in base_model.layers:
layer.trainable = False
model.compile(optimizer=adam(lr=0.01), loss='binary_crossentropy', metrics=['accuracy'])
return model
class Histories(keras.callbacks.Callback):
def __init__(self, val_data):
super(Histories, self).__init__() …Run Code Online (Sandbox Code Playgroud)