def returnTF(): # load the image transformer tf = trn.Compose([ trn.Scale((224,224)), trn.ToTensor(), trn.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) ]) return tf