add the version that resize the image instead of padding
put the testing stage to CPU instead of GPU