using transform to norm image to (256,256).
batch size parameterized.