PyTorch深度学习实战(15)——迁移学习

    • 0. 前言
    • 1. 迁移学习
      • 1.1 迁移学习基本概念
      • 1.2 迁移学习的重要性
      • 1.3 ImageNet
      • 1.4 迁移学习流程
    • 2. VGG16 架构
    • 3. 使用预训练 VGG16 模型实现猫狗分类
    • 小结
    • 系列链接

0. 前言

迁移学习( Transfer Learning )是一种利用从一项任务中获得的知识来解决另一项类似任务的技术。一个使用数百万张图像训练的模型,训练数据涵盖数千种对象类别,模型的卷积核将能够学习图像中的各种形状、颜色和纹理,通过重用这些卷积核可以学习到新图像的特征,并最终用于执行计算机视觉任务。随着训练数据集中可用图像数量的增加,模型的分类准确率会不断提高,然而,在实际训练模型过程中,获取大量具有标签的数据样本通常比较困难,需要耗费大量时间和人力成本,而迁移学习能够在训练数据不足的情况下实现更好的泛化性能。

1. 迁移学习