Modify Python Class to filter files

You could e.g. create a custom find_classes method and pass it to DatasetFolder as described here.