The Dataset Class
Learn about the dataset class.
We'll cover the following
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 1200+ tech skills courses.