【发布时间】:2018-01-05 09:30:43
【问题描述】:
所以我在一个小数据集上进行二值图像分类每个类别包含 250 个图像,我使用迁移学习,使用 Resnet50 作为基础网络架构并在它之上我添加了 2 个隐藏层和 1 个最终输出层,经过 20 个 epoch 的训练,我看到损失在初始 epoch 突然增加,我无法理解其背后的原因。
网络架构 -
image_input = Input(shape=(224, 224, 3))
model = ResNet50(input_tensor=image_input,include_top=True, weights='imagenet')
last_layer = model.get_layer('avg_pool').output
x = Flatten(name='flatten')(last_layer)
x = Dense(1000, activation='relu', name='fc1000')(x)
x = Dropout(0.5)(x)
x = Dense(200, activation='relu', name='fc200')(x)
x = Dropout(0.5)(x)
out = Dense(num_classes, activation='softmax', name='output')(x)
custom_model = Model(image_input, out)
我正在使用 binary_crossentropy,Adam 和默认参数
【问题讨论】:
标签: python keras conv-neural-network resnet