The Dataset Class

Learn about 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.

Get hands-on with 1400+ tech skills courses.