torchvision源代码浅析【2】
本文以mnist介绍dateset,并介绍了如何自定义dataset。
mnist代码: 下载
可以看到,一个dataset都要继承一个torch.utils.data内的Dataset类。为了使一个dataset可以被pytorch内置的dataloader载入,必须实现两个method:
def __getitem__(self, index):
def __len__(self):
第一个__getitem__通过一个index来返回一个数据项,第二个__len__返回数据的数目。