Какие параметры надо передать в transforms.Normalize()?

Нашел готовый код на pytorch, там:

torchvision.transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

Но согласно документации числа, которые мы передаем, должны быть вычислены предварительно для нашего набора данных.

Как вычислить эти числа?


Ответы (1 шт):

Автор решения: MaxU

Среднее и стандартное отклонение высчитывается для всего набора картинок. Значения из вашего вопроса - это среднее и отклонение для всего набора ImageNet. Для того, чтобы посчитать данные значения для вашего набора данных нужно посчитать среднее и стандартное отклонение для всех картинок из вашего набора данных.

→ Ссылка