python深度学习 微调模型与冻结部分层

       特征微调有助于提高深度学习的精度,而将卷积基(就是预先训练好的卷积模型)的前面大部分层冻结,只微调(此时的学习率也应该相应的调低)顶部更抽象的层---可以理解为更细节的特征

这样可以在一定程度上提高模型的识别精度,我的实验结果比原有的精度提高了4个点左右,从90%提高到94%

下面是全部的代码,很简单.

不过有个疑问:训练出来的验证精度达不到书上说的97%,有解决的同学请留言告知一下,不慎感激!!!

#即可单独安装keras,也可以用tensorflow自带的keras
try:
    from keras.applications import VGG16
    from keras.preprocessing.image import ImageDataGenerator
    from keras import models
    from keras import layers
    from keras import optimizers
except ModuleNotFoundError:
    import tensorflow as tf
    VGG16 = tf.keras.applications.VGG16
    ImageDataGenerator = tf.keras.preprocessing.image.ImageDataGenerator
    models = tf.keras.models
    layers = tf.keras.layers
    optimizers = tf.keras.optimizers
import os,sys
sys.path.append( os.pardir )  #为了找到上级目录common,import里面的工具
from common.file_arrange import get_dirs  #自定义的小工具
import numpy as np 
import matplotlib.pyplot as plt 

def get_co
  • 5
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值