I am doing some experiments on MNIST. I want to preprocess images before feed them into CNN. My preprocess function is like
def preprocess(image):
...
return processed_image
the result is a processed image with the same size as the original image.
I am trying to modify this code: pytorch examples mnist
I am wondering where I should add this preprocess function into the code. Thank you!