调用resnet50权重_无法使用ResNet50在Keras中加载用于微调的权重

在尝试使用ResNet50预训练模型进行微调时遇到错误,模型加载权重过程中出现形状不匹配的问题。当切换到VGG19时,问题消失,引发对于Keras中ResNet实现的疑问。
摘要由CSDN通过智能技术生成

我首先使用ResNet-50层在我的数据集上冻结,使用以下方法进行培训:model_r50 = ResNet50(weights='imagenet', include_top=False)

model_r50.summary()

input_layer = Input(shape=(img_width,img_height,3),name = 'image_input')

output_r50 = model_r50(input_layer)

fl = Flatten(name='flatten')(output_r50)

dense = Dense(1024, activation='relu', name='fc1')(fl)

drop = Dropout(0.5, name='drop')(dense)

pred = Dense(nb_classes, activation='softmax', name='predictions')(drop)

fine_model = Model(outputs=pred,inputs=input_layer)

for layer in model_r50.layers:

layer.trainable = False

print layer

fine_model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])

fine_

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值