Thank you for your reply!
I was unaware that it is always recommended to use Custom Datasets when not using a PyTorch dataset.
After creating the custom dataset class and instantiating it, I have maintained the desired dimensions, but the images are losing the 3 RGB channels and the images being converted to binary for some reason. I think it’s due to my transforms, but still figuring that out.