The Dataset Class

To work with images from the CelebA dataset we’ll need to modify the Dataset class that we used with the MNIST images.

The CelebADataset class

The changes are fairly simple because we’ve already seen how to open a HDF5 file and extract images from it. Have a look at this CelebADataset class.