├── .gitignore ├── ReadME.md ├── data ├── TransformersDataset.py ├── __init__.py └── samplers.py ├── fig ├── MAE.png ├── decoder.png └── decoder_vitbase.png ├── inference.py ├── loss ├── __int__.py ├── centerloss.py ├── focalloss.py └── mae_loss.py ├── mae_test.py ├── model ├── Transformers │ ├── VIT │ │ ├── __init__.py │ │ ├── demo.py │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── drop.py │ │ │ ├── helpers.py │ │ │ ├── mlp.py │ │ │ ├── patch_embd.py │ │ │ └── weight_init.py │ │ ├── mae.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ └── mask_embeeding.py │ │ └── vit.py │ └── __init__.py └── __init__.py ├── train_mae.py ├── train_mae.sh └── utils ├── EMA.py ├── LARS.py ├── SWA.py ├── __init__.py ├── augments.py ├── data_aug.py ├── labelsmooth.py ├── mixup.py ├── optim_factory.py ├── optimizer_step.py ├── precise_bn.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /ReadME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/ReadME.md -------------------------------------------------------------------------------- /data/TransformersDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/data/TransformersDataset.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/data/samplers.py -------------------------------------------------------------------------------- /fig/MAE.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/fig/MAE.png -------------------------------------------------------------------------------- /fig/decoder.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/fig/decoder.png -------------------------------------------------------------------------------- /fig/decoder_vitbase.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/fig/decoder_vitbase.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/inference.py -------------------------------------------------------------------------------- /loss/__int__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /loss/centerloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/loss/centerloss.py -------------------------------------------------------------------------------- /loss/focalloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/loss/focalloss.py -------------------------------------------------------------------------------- /loss/mae_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/loss/mae_loss.py -------------------------------------------------------------------------------- /mae_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/mae_test.py -------------------------------------------------------------------------------- /model/Transformers/VIT/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Transformers/VIT/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/demo.py -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/layers/drop.py -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/layers/helpers.py -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/layers/mlp.py -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/patch_embd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/layers/patch_embd.py -------------------------------------------------------------------------------- /model/Transformers/VIT/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/layers/weight_init.py -------------------------------------------------------------------------------- /model/Transformers/VIT/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/mae.py -------------------------------------------------------------------------------- /model/Transformers/VIT/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Transformers/VIT/utils/mask_embeeding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/utils/mask_embeeding.py -------------------------------------------------------------------------------- /model/Transformers/VIT/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/model/Transformers/VIT/vit.py -------------------------------------------------------------------------------- /model/Transformers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/train_mae.py -------------------------------------------------------------------------------- /train_mae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/train_mae.sh -------------------------------------------------------------------------------- /utils/EMA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/EMA.py -------------------------------------------------------------------------------- /utils/LARS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/LARS.py -------------------------------------------------------------------------------- /utils/SWA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/SWA.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/augments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/augments.py -------------------------------------------------------------------------------- /utils/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/data_aug.py -------------------------------------------------------------------------------- /utils/labelsmooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/labelsmooth.py -------------------------------------------------------------------------------- /utils/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/mixup.py -------------------------------------------------------------------------------- /utils/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/optim_factory.py -------------------------------------------------------------------------------- /utils/optimizer_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/optimizer_step.py -------------------------------------------------------------------------------- /utils/precise_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/MAE-pytorch/HEAD/utils/precise_bn.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------