I would like to perform different transformations on  5 different classes in a multi-class classification problem. What’s the cleanest way to do this using ImageFolder and DataLoader?
Note: The classes also have quite different sizes, ranging from 50-3,000 images
How can I set this up to not over fit on the classes with more data?
My data structure looks like this:
/data
    /train
        /class1
        /class2
        /class3
        /class4
        /class5
    /test
        /class1
        /class2
        /class3
        /class4
        /class5