DATASET CLASS
- 自定义的dataset类需要继承torch.utils.data.Dataset。并且实现两个函数:len、getitem。
- 注意FaceLandmarksDataset只将landmarks存储到内存中,image是lazy-loading的机制。
TRANSFORMS
ITERATING THROUGH THE DATASET
- torch.utils.data.DataLoader可以补充三个功能:batch the data、shuffling the data、load the data in parallel using multiprocessing workers.它是一个iterator。