pytorch学习笔记-----分类网络实践

基于经典网络训练图像分类

数据预处理部分：

        可以利用torchvision.transforms模块进行数据增强，及数据预处理

        制作可以读取数据的dataloader

网络模型设置：

        可以加载预训练模型，例如torchvision经典网络框架

        可以更改最后的head层训练自己的任务，大多为全连接层

        前几层多为特征提取我们可以只训练最后任务层

网络模型保存