├── .gitignore ├── LICENSE ├── README.md ├── docs ├── FINETUNE.md ├── INSTALL.md ├── PRETRAIN.md └── teaser.jpg ├── engines ├── finetune.py ├── pretrain_cim.py └── pretrain_mae.py ├── models ├── cim.py ├── mae.py ├── nn_util.py └── vit.py ├── requirements.txt ├── scripts ├── dist_finetune.sh └── dist_pretrain.sh ├── tools ├── main_finetune.py └── main_pretrain.py └── util ├── crop.py ├── datasets.py ├── lars.py ├── lr_decay.py ├── lr_sched.py ├── misc.py └── pos_embed.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/README.md -------------------------------------------------------------------------------- /docs/FINETUNE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/docs/FINETUNE.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/PRETRAIN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/docs/PRETRAIN.md -------------------------------------------------------------------------------- /docs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/docs/teaser.jpg -------------------------------------------------------------------------------- /engines/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/engines/finetune.py -------------------------------------------------------------------------------- /engines/pretrain_cim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/engines/pretrain_cim.py -------------------------------------------------------------------------------- /engines/pretrain_mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/engines/pretrain_mae.py -------------------------------------------------------------------------------- /models/cim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/models/cim.py -------------------------------------------------------------------------------- /models/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/models/mae.py -------------------------------------------------------------------------------- /models/nn_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/models/nn_util.py -------------------------------------------------------------------------------- /models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/models/vit.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | timm==0.5.4 2 | pillow==9.0.1 -------------------------------------------------------------------------------- /scripts/dist_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/scripts/dist_finetune.sh -------------------------------------------------------------------------------- /scripts/dist_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/scripts/dist_pretrain.sh -------------------------------------------------------------------------------- /tools/main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/tools/main_finetune.py -------------------------------------------------------------------------------- /tools/main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/tools/main_pretrain.py -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/lars.py -------------------------------------------------------------------------------- /util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/lr_decay.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weivision/Correlational-Image-Modeling/HEAD/util/pos_embed.py --------------------------------------------------------------------------------