├── .gitignore ├── README.md ├── configs └── imagenet.yaml ├── datasets.py ├── evaluations ├── finetune.py └── linear_prob.py ├── mae.py ├── main.py ├── requirements.txt ├── scripts └── pretrain_and_finetune.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .vscode 2 | .DS_Store 3 | __pycache__ -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/README.md -------------------------------------------------------------------------------- /configs/imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/configs/imagenet.yaml -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/datasets.py -------------------------------------------------------------------------------- /evaluations/finetune.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluations/linear_prob.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/mae.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/main.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/pretrain_and_finetune.sh: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PatrickHua/SimpleMAE/HEAD/utils.py --------------------------------------------------------------------------------