zwx8981 / UNIQUE

The repository for 'Uncertainty-aware blind image quality assessment in the laboratory and wild' and 'Learning to blindly assess image quality in the laboratory and wild'
Apache License 2.0
127 stars 10 forks source link

Normalization for testing images #7

Closed pencilzhang closed 3 years ago

pencilzhang commented 3 years ago

Hi,

I am wondering why you do normalization on testing images? transforms.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225))])

zwx8981 commented 3 years ago

@pencilzhang We need to do the same normalization between training and testing phases.