迁移学习中解决keras模型训练过程中的验证集准确率不变问题

方案一:

设置bn层参数不可变

K.set_learning_phase(0)
base_model = tf.keras.applications.ResNet50(weights='imagenet', include_top=False,
                                  input_shape=(img_resize, img_resize, 3), )
# 设置bn层参数可变
K.set_learning_phase(1)
model=tf.keras.Sequential()
model.add(base_model)
model.add(tf.keras.layers.GlobalAveragePooling2D())
model.add(tf.keras.layers.Dense(512,activation="relu"))
#model.add(tf.keras.layers.Dense(512,activation="relu",kernel_regularizer=tf.keras.regularizers.l2(0.0001)))
model.add(tf.keras.layers.Dense(1,activation="sigmoid"))
model.compile(optimizer=tf.keras.optimizers.Adam(lr=0.0001),loss="binary_crossentropy",metrics=["acc"])
history=model.fit(train_ds2,epochs=5,validation_data=test_ds2)

方案二:

K.set_learning_phase(0)
base_model = tf.keras.applications.ResNet50(weights='imagenet', include_top=False,input_shape=(img_resize, img_resize, 3) )
# 不要设置成False,否则会影响训练效果(冻结训练的代码一定要在这个位置写)
base_model.trainable=False
K.set_learning_phase(1)
x=base_model.output
x=tf.keras.layers.GlobalAveragePooling2D()(x)
x=tf.keras.layers.Dense(512,activation="relu")(x)
x=tf.keras.layers.Dense(1,activation="sigmoid")(x)
model = tf.keras.Model(inputs=base_model.input, outputs=x)
  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值