Closed RookieHong closed 3 years ago
The images should be normalized to [0, 1], what about the mean and std? For example, the normalization for pytorch pre-trained models is: normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
The model has been trained with inputs simply scaled in [0, 1].
You mean, just simply divide the pixel values by 255?
right! it should be easier than the usual pytorch normalization :D
Thanks!
The images should be normalized to [0, 1], what about the mean and std? For example, the normalization for pytorch pre-trained models is:
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])