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__返回数据的数目。