PyTorch教程:21个项目玩转PyTorch实战
上QQ阅读APP看书,第一时间看更新

第3章 从零构建图像分类模型

本章着重介绍如何使用 PyTorch 在自己收集的图像数据集上训练深度学习模型,主要方法是使用视觉工具包torchvision中提供的在 ImageNet 上预训练好的图像分类模型,分别介绍以微调的方式和将其作为特征提取器的方式来构建自己的深度学习模型。