How can I obtain a single or required number of image from the vision dataset for each class of images?

I want to display 2-3 images from each class in my dataset. However, when I load the dataset, it downloads the entire dataset. I only need 2-3 images from each class. How can I achieve this?

You can use streaming (to avoid downloading the entire dataset) and iterate over the dataset until you obtain the number of images you want from each class.