dingkeyan93 / DISTS

IQA: Deep Image Structure and Texture Similarity Metric
MIT License
359 stars 42 forks source link

pytorch pretrained vgg #5

Open davidebaltieri31 opened 3 years ago

davidebaltieri31 commented 3 years ago

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.

zakajd commented 3 years ago

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