├── LICENSE ├── README.md ├── dataset_folder.py ├── datasets.py ├── engine_for_finetuning.py ├── engine_for_pretraining.py ├── imagefromlist.py ├── imgs ├── framework.png ├── results.png └── vis.png ├── masking_generator.py ├── modeling_finetune.py ├── modeling_pretrain.py ├── optim_factory.py ├── requirements.txt ├── run_class_finetuning.py ├── run_conmim_pretraining.py ├── transforms.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/README.md -------------------------------------------------------------------------------- /dataset_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/dataset_folder.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/datasets.py -------------------------------------------------------------------------------- /engine_for_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/engine_for_finetuning.py -------------------------------------------------------------------------------- /engine_for_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/engine_for_pretraining.py -------------------------------------------------------------------------------- /imagefromlist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/imagefromlist.py -------------------------------------------------------------------------------- /imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/imgs/framework.png -------------------------------------------------------------------------------- /imgs/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/imgs/results.png -------------------------------------------------------------------------------- /imgs/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/imgs/vis.png -------------------------------------------------------------------------------- /masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/masking_generator.py -------------------------------------------------------------------------------- /modeling_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/modeling_finetune.py -------------------------------------------------------------------------------- /modeling_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/modeling_pretrain.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/optim_factory.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_class_finetuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/run_class_finetuning.py -------------------------------------------------------------------------------- /run_conmim_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/run_conmim_pretraining.py -------------------------------------------------------------------------------- /transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/transforms.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TencentARC/ConMIM/HEAD/utils.py --------------------------------------------------------------------------------