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

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

数据预处理部分:

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

        制作可以读取数据的dataloader

网络模型设置:

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

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

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

网络模型保存

你可能感兴趣的:(pytorch学习笔记,网络,人工智能,深度学习,机器学习,神经网络)