自定义数据集
要想使用自己的数据集,必须实现3个方法: __init__()
,__len__()
与__getitem__()
.
___init__()
:
在初始化数据集时调用一次,我们在初始化方法里完成一些数据的初始化工作,例如,初始化图片路径,标注文件以及数据预处理的初始化。
__len__()
:
返回数据集中的数据数量
__getitem__()
:
__getitem__()
的方法根据给定的idx,加载然后返回一个样本。基于idx,找到图片并转换成tensor.遍历label,调用自定义的transform函数来完成图片的预处理。最终返回tensor化的图片和对应的标签元组。
【Pytorch】学习笔记(一)创建自己的数据集
于 2022-07-19 09:20:34 首次发布