eat_pytorch_in_20_days学习笔记(1)图片数据建模流程范例
一、Pytorch的建模流程
1.准备数据(难点。我的数据类型:图片数据)
2.定义模型
3.训练模型
4.评估模型
5.使用模型
6.保存模型
图片数据建模流程范例
1.准备数据
cifar2数据集:
cifar10数据集的子集,只包括前两种类别airplane和automobile
训练集:
airplane和automobile图片各5000张
测试集:
airplane和automobile图片各1000张
任务的目标:
训练一个模型,来对飞机airplane和机动车automobile两种图片进行
原创
2020-06-30 19:14:32 ·
519 阅读 ·
0 评论