当前位置:   article > 正文

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

深度学习冷冻和微调

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

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

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

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

  1. #即可单独安装keras,也可以用tensorflow自带的keras
  2. try:
  3. from keras.applications import VGG16
  4. from keras.preprocessing.image import ImageDataGenerator
  5. from keras import models
  6. from keras import layers
  7. from keras import optimizers
  8. except ModuleNotFoundError:
  9. import tensorflow as tf
  10. VGG16 = tf.keras.applications.VGG16
  11. ImageDataGenerator = tf.keras.preprocessing.image.ImageDataGenerator
  12. models = tf.keras.models
  13. layers = tf.keras.layers
  14. optimizers = tf.keras.optimizers
  15. import os,sys
  16. sys.path.append( os.pardir ) #为了找到上级目录common,import里面的工具
  17. from common.file_arrange import get_dirs #自定义的小工具
  18. import numpy as np
  19. import matplotlib.pyplot as plt
  20. def get_co
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/101258
推荐阅读
相关标签
  

闽ICP备14008679号