input image is first loaded to range [0, 1] and then this normalization is applied to RGB image as described here:

  
      
      GitHub
  
  
    

pytorch/vision 320

Datasets, Transforms and Models specific to Computer Vision - pytorch/vision


  
  
    
    
  
  



All pre-trained models expect input images normalized in the same way, i.e. mini-batches of 3-channel RGB images of shape (3 x H x W), where H and W are expected to be atleast 224.
The images have to be loaded in to a range of [0, 1] and then normalized using mean=[0.485, 0.456, 0.406] and std=[0.229, 0.224, 0.225]
An example of such normalization can be found in the imagenet example here https://github.com/pytorch/examples/blob/42e5b996718797e45c46a25c55b031e6768f8440/imagenet/main.py#L89-L101 1.3k