Thank you , but the machine learning engineer told me it’s impossible to add the preprocess into the model. I reviewed the code and then find it doesn’t have the code (No normalize), does it influence the effect?
We compose a sequence of transformation to pre-process the image:
import torchvision.transforms as transforms

transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])