Is there a way of getting images of a certain class type, e.g. bengal cat, pets dataset, from a dataloader?