参考视频:
P5. PyTorch加载数据初认识_哔哩哔哩_bilibili
P6. Dataset类代码实战_哔哩哔哩_bilibili
1. dataset
1.1. 作用:
- 获取数据&其label
- 告知共有多少数据
1.2. 实战
1.2.1. 数据集准备
下载数据集:https://download.pytorch.org/tutorial/hymenoptera_data.zip
该数据集是蚂蚁和蜜蜂的二分类数据集,label就是文件名字,数据集结构如下
hymenoptera_data
|__train
| |__ants
| |__bees
|
|__val