DISTS icon indicating copy to clipboard operation
DISTS copied to clipboard

pytorch pretrained vgg

Open davidebaltieri31 opened this issue 5 years ago • 1 comments

Not an issue per se, more like a suggestion.

I see you use pytorch pretrained vgg without correctly normalizing the image (all pytorch pretrained models assume the image was normalized with transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) )

Retraining with the correct normalization might give you even better results.

davidebaltieri31 avatar Nov 13 '20 11:11 davidebaltieri31

They actually do normalize images in forward_once function, using self.mean and self.std constants

jzakirov avatar Jan 12 '21 17:01 jzakirov