Currently you are just passing the module transforms to your Dataset. Instead you should specify, which transformation you would like to be applied on your data, e.g. transforms.Resize(224). You can find the transformations in the docs 31. If you find some interesting image transformations, you would have to cast your numpy array to a PIL.Image before applying them.