The easiest way that comes to my mind at the moment is to create a new Dataset instance and explicitly use different transformations on different classes.
For the unbalanced dataset, you may want to use WeightedRandomSampler