Well, I’m not sure if I understood what you are willing to do but, as far as I know, the Pytorch dataset is making the transformation while you are training.
Which means, for now, non of your images where transformed.
But, you can see the size of your dataset by calling the method len()
in your code: train_data. _ _len _ _ ()
Regardless to that, you can iterate over the train/test loaders, you will get number of values and labels as your batch size after the transformation. It is like seeing how your data will be feed into your model.
If your dataset only returns x (value) and y (label). you can write this:
x,y = iter(train_loader).next()
both will be in the size of your batch size.
x will contain your input tensors after the transformation
y will contain the labels of these inputs
after that you can access to one of your inputs by writing: x[0] (or other number in the range of your batch size)