【B站】小土堆pytorch-CSDN博客
介绍
这里用垃圾分类来处理。
Dataset: 提供一种方法把数据加载进来,提供两个方法,分别实现获取数据和label、获取数据总量的功能。
Dataloader: 把数据打包交给模型训练、具体后面再提。
代码实战
这里用了一个二分类数据集:ants和bees
包含图片及其label。
这里用垃圾分类来处理。
Dataset: 提供一种方法把数据加载进来,提供两个方法,分别实现获取数据和label、获取数据总量的功能。
Dataloader: 把数据打包交给模型训练、具体后面再提。
这里用了一个二分类数据集:ants和bees
包含图片及其label。