TensorFlow2.0.0_课时45-49_迁移学习

本文介绍了在TensorFlow2.0.0中使用迁移学习的方法,特别是通过ResNet模型进行特征提取。讨论了不同数据量下微调策略,如数据量小时冻结大部分层,数据量大时允许部分层训练。同时,讲解了如何利用预训练模型的权重作为初始化,并探讨了callback在训练过程中的作用,包括早停、学习率调整和模型保存等功能。最后,展示了如何选择模型中的特定层进行训练和绘图分析。
摘要由CSDN通过智能技术生成

特征提取尽量微调
全连接层自己做
数据量小,全冻
数据量中,10000左右,冻前边的
数据量大,冻的少
Resnet论文:Deep Residual Learnling for Image Recognition,Kaiming He

tensorflow.keras.applications
里面有很多现成训练好的模型可以直接利用
如:tf.keras.applications.ResNet152

迁移学习
用人家训练好模型的权重参数当做咱们的初始化
一般全连接层需要自己训练,可以选择是否训练别人训练好的特征提取层

主代码

import os
import warnings
warnings.filterwarnings("ignore")
import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow.keras.optimizers import Adam
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras import layers
from tensorflow.keras import Model
base_dir = './data/cats_and_dogs'
train_dir = os.path.join(base_dir, 'train')
validation_dir = os.path.join(base_dir, 'validation')

train_cats_dir = os.path.join(train_dir, 'cats')
train_dogs_dir = os.path.join(train_dir, 'dogs')

validation_cats_dir = os.path.join(validation_dir, 'cats')
validation_dogs_dir = os.path.join(validation_dir, 'dogs')
### 导入模型
from tf.keras.applications.resnet import ResNet50
from tensorflow.keras.applications.
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值