上传者: 38684892
|
上传时间: 2021-12-30 22:00:58
|
文件大小: 785KB
|
文件类型: -
自定义数据集
在训练深度学习模型之前,样本集的制作非常重要。在pytorch中,提供了一些接口和类,方便我们定义自己的数据集合,下面完整的试验自定义样本集的整个流程。
开发环境
Ubuntu 18.04
pytorch 1.0
pycharm
实验目的
掌握pytorch中数据集相关的API接口和类
熟悉数据集制作的整个流程
实验过程
1.收集图像样本
以简单的猫狗二分类为例,可以在网上下载一些猫狗图片。创建以下目录:
data————-根目录
data/test——-测试集
data/train——训练集
data/val——–验证集
在test/tra