Error in pre-processing of the test_images

Refer to Inferring resnet18 classification etlt model with python - #12 by Morganh