antoine77340 / S3D_HowTo100M

S3D Text-Video model trained on HowTo100M using MIL-NCE
Apache License 2.0
191 stars 21 forks source link

How to normalize the images? #10

Closed RookieHong closed 3 years ago

RookieHong commented 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])

antoine77340 commented 3 years ago

The model has been trained with inputs simply scaled in [0, 1].

RookieHong commented 3 years ago

You mean, just simply divide the pixel values by 255?

antoine77340 commented 3 years ago

right! it should be easier than the usual pytorch normalization :D

RookieHong commented 3 years ago

Thanks!