Allow users to choose the imagenet mean in config file

BR-IDL / PaddleViT

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

https://github.com/BR-IDL/PaddleViT

Apache License 2.0

1.22k stars 318 forks source link

Allow users to choose the imagenet mean in config file #106

Closed xperzy closed 2 years ago

xperzy commented 2 years ago

Describe your feature request Now dataset.py has hard coded imagenet mean and var, which is not flexible and not easy to find for new users. This can be set with an argument in config file, users can set the default mean, var values.