├── .DS_Store ├── .gitignore ├── DET ├── DETECTION.md ├── configs │ ├── coco.py │ ├── common.py │ └── convmae │ │ ├── convmae_base_mask_rcnn_FPN_100ep.py │ │ ├── convmae_base_mask_rcnn_FPN_25ep.py │ │ └── convmae_base_mask_rcnn_FPN_50ep.py ├── lazyconfig_train_net.py ├── models │ ├── __init__.py │ ├── convmae.py │ └── modeling │ │ ├── __init__.py │ │ ├── postprocessing.py │ │ └── rcnn.py └── utils │ ├── __init__.py │ └── pos_embed.py ├── FINETUNE.md ├── LICENSE ├── PRETRAIN.md ├── README.md ├── SEG ├── SEGMENTATION.md ├── backbone │ └── convmae.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── ade20k.py │ │ │ └── ade20k_640x640.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ └── upernet.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ └── schedule_320k.py │ └── convmae │ │ └── upernet_convmae_base_512_slide_160k_ade20k.py ├── mmcv_custom │ ├── __init__.py │ ├── apex_runner │ │ ├── __init__.py │ │ ├── apex_iter_based_runner.py │ │ ├── checkpoint.py │ │ └── optimizer.py │ ├── checkpoint.py │ ├── layer_decay_optimizer_constructor.py │ ├── resize_transform.py │ └── train_api.py └── tools │ ├── dist_test.sh │ ├── dist_train.sh │ ├── flops.py │ ├── test.py │ └── train.py ├── engine_finetune.py ├── engine_pretrain.py ├── figures ├── ConvMAE.png ├── Downstream.png └── feat_map.JPG ├── main_finetune.py ├── main_linprobe.py ├── main_pretrain.py ├── models_convmae.py ├── models_convvit.py ├── submitit_finetune.py ├── submitit_linprobe.py ├── submitit_pretrain.py ├── util ├── crop.py ├── datasets.py ├── lars.py ├── lr_decay.py ├── lr_sched.py ├── misc.py └── pos_embed.py └── vision_transformer.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/.gitignore -------------------------------------------------------------------------------- /DET/DETECTION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/DETECTION.md -------------------------------------------------------------------------------- /DET/configs/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/configs/coco.py -------------------------------------------------------------------------------- /DET/configs/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/configs/common.py -------------------------------------------------------------------------------- /DET/configs/convmae/convmae_base_mask_rcnn_FPN_100ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/configs/convmae/convmae_base_mask_rcnn_FPN_100ep.py -------------------------------------------------------------------------------- /DET/configs/convmae/convmae_base_mask_rcnn_FPN_25ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/configs/convmae/convmae_base_mask_rcnn_FPN_25ep.py -------------------------------------------------------------------------------- /DET/configs/convmae/convmae_base_mask_rcnn_FPN_50ep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/configs/convmae/convmae_base_mask_rcnn_FPN_50ep.py -------------------------------------------------------------------------------- /DET/lazyconfig_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/lazyconfig_train_net.py -------------------------------------------------------------------------------- /DET/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/models/__init__.py -------------------------------------------------------------------------------- /DET/models/convmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/models/convmae.py -------------------------------------------------------------------------------- /DET/models/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/models/modeling/__init__.py -------------------------------------------------------------------------------- /DET/models/modeling/postprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/models/modeling/postprocessing.py -------------------------------------------------------------------------------- /DET/models/modeling/rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/models/modeling/rcnn.py -------------------------------------------------------------------------------- /DET/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /DET/utils/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/DET/utils/pos_embed.py -------------------------------------------------------------------------------- /FINETUNE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/FINETUNE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/LICENSE -------------------------------------------------------------------------------- /PRETRAIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/PRETRAIN.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/README.md -------------------------------------------------------------------------------- /SEG/SEGMENTATION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/SEGMENTATION.md -------------------------------------------------------------------------------- /SEG/backbone/convmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/backbone/convmae.py -------------------------------------------------------------------------------- /SEG/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /SEG/configs/_base_/datasets/ade20k_640x640.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/datasets/ade20k_640x640.py -------------------------------------------------------------------------------- /SEG/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /SEG/configs/_base_/models/upernet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/models/upernet.py -------------------------------------------------------------------------------- /SEG/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /SEG/configs/_base_/schedules/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/_base_/schedules/schedule_320k.py -------------------------------------------------------------------------------- /SEG/configs/convmae/upernet_convmae_base_512_slide_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/configs/convmae/upernet_convmae_base_512_slide_160k_ade20k.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/__init__.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/apex_runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/apex_runner/__init__.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/apex_runner/apex_iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/apex_runner/apex_iter_based_runner.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/apex_runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/apex_runner/checkpoint.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/apex_runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/apex_runner/optimizer.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/checkpoint.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/layer_decay_optimizer_constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/layer_decay_optimizer_constructor.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/resize_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/resize_transform.py -------------------------------------------------------------------------------- /SEG/mmcv_custom/train_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/mmcv_custom/train_api.py -------------------------------------------------------------------------------- /SEG/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/tools/dist_test.sh -------------------------------------------------------------------------------- /SEG/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/tools/dist_train.sh -------------------------------------------------------------------------------- /SEG/tools/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/tools/flops.py -------------------------------------------------------------------------------- /SEG/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/tools/test.py -------------------------------------------------------------------------------- /SEG/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/SEG/tools/train.py -------------------------------------------------------------------------------- /engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/engine_finetune.py -------------------------------------------------------------------------------- /engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/engine_pretrain.py -------------------------------------------------------------------------------- /figures/ConvMAE.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/figures/ConvMAE.png -------------------------------------------------------------------------------- /figures/Downstream.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/figures/Downstream.png -------------------------------------------------------------------------------- /figures/feat_map.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/figures/feat_map.JPG -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/main_finetune.py -------------------------------------------------------------------------------- /main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/main_linprobe.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /models_convmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/models_convmae.py -------------------------------------------------------------------------------- /models_convvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/models_convvit.py -------------------------------------------------------------------------------- /submitit_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/submitit_finetune.py -------------------------------------------------------------------------------- /submitit_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/submitit_linprobe.py -------------------------------------------------------------------------------- /submitit_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/submitit_pretrain.py -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/lars.py -------------------------------------------------------------------------------- /util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/lr_decay.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/util/pos_embed.py -------------------------------------------------------------------------------- /vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Alpha-VL/ConvMAE/HEAD/vision_transformer.py --------------------------------------------------------------------------------