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



