├── .gitignore ├── LICENSE ├── README.md ├── configs ├── few-shot │ ├── dist_fewshot_sim_base.sh │ └── slurm_fewshot_sim_base.sh ├── finetune │ ├── dist_finetune_sim_base.sh │ ├── dist_finetune_sim_base_eval.sh │ ├── slurm_finetune_sim_base.sh │ └── slurm_finetune_sim_base_eval.sh ├── linprobe │ ├── dist_linprobe_sim_base.sh │ └── slurm_linprobe_sim_base.sh ├── pretrain │ ├── dist_sim_base_1600ep.sh │ └── slurm_sim_base_1600ep.sh ├── semisup_rebuttal.sh ├── semisup_sim_base_400ep.sh └── semisup_sim_large_1600ep.sh ├── docs ├── checkpoints.md ├── few_shot.md ├── finetune.md ├── linear_eval.md ├── prepare.md └── pretrain.md ├── engine_finetune.py ├── engine_pretrain.py ├── figs ├── comparison.png └── overview.png ├── imagenet_subset1 └── 1percent.txt ├── main_finetune.py ├── main_linprobe.py ├── main_logistic.py ├── main_pretrain.py ├── models_sim.py ├── models_vit.py └── util ├── augmentation.py ├── crop.py ├── datasets.py ├── lars.py ├── lr_decay.py ├── lr_sched.py ├── masking_generator.py ├── misc.py ├── pos_embed.py └── tcs_datasets.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/README.md -------------------------------------------------------------------------------- /configs/few-shot/dist_fewshot_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/few-shot/dist_fewshot_sim_base.sh -------------------------------------------------------------------------------- /configs/few-shot/slurm_fewshot_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/few-shot/slurm_fewshot_sim_base.sh -------------------------------------------------------------------------------- /configs/finetune/dist_finetune_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/finetune/dist_finetune_sim_base.sh -------------------------------------------------------------------------------- /configs/finetune/dist_finetune_sim_base_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/finetune/dist_finetune_sim_base_eval.sh -------------------------------------------------------------------------------- /configs/finetune/slurm_finetune_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/finetune/slurm_finetune_sim_base.sh -------------------------------------------------------------------------------- /configs/finetune/slurm_finetune_sim_base_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/finetune/slurm_finetune_sim_base_eval.sh -------------------------------------------------------------------------------- /configs/linprobe/dist_linprobe_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/linprobe/dist_linprobe_sim_base.sh -------------------------------------------------------------------------------- /configs/linprobe/slurm_linprobe_sim_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/linprobe/slurm_linprobe_sim_base.sh -------------------------------------------------------------------------------- /configs/pretrain/dist_sim_base_1600ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/pretrain/dist_sim_base_1600ep.sh -------------------------------------------------------------------------------- /configs/pretrain/slurm_sim_base_1600ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/pretrain/slurm_sim_base_1600ep.sh -------------------------------------------------------------------------------- /configs/semisup_rebuttal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/semisup_rebuttal.sh -------------------------------------------------------------------------------- /configs/semisup_sim_base_400ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/semisup_sim_base_400ep.sh -------------------------------------------------------------------------------- /configs/semisup_sim_large_1600ep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/configs/semisup_sim_large_1600ep.sh -------------------------------------------------------------------------------- /docs/checkpoints.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/checkpoints.md -------------------------------------------------------------------------------- /docs/few_shot.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/few_shot.md -------------------------------------------------------------------------------- /docs/finetune.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/finetune.md -------------------------------------------------------------------------------- /docs/linear_eval.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/linear_eval.md -------------------------------------------------------------------------------- /docs/prepare.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/prepare.md -------------------------------------------------------------------------------- /docs/pretrain.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/docs/pretrain.md -------------------------------------------------------------------------------- /engine_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/engine_finetune.py -------------------------------------------------------------------------------- /engine_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/engine_pretrain.py -------------------------------------------------------------------------------- /figs/comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/figs/comparison.png -------------------------------------------------------------------------------- /figs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/figs/overview.png -------------------------------------------------------------------------------- /imagenet_subset1/1percent.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/imagenet_subset1/1percent.txt -------------------------------------------------------------------------------- /main_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/main_finetune.py -------------------------------------------------------------------------------- /main_linprobe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/main_linprobe.py -------------------------------------------------------------------------------- /main_logistic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/main_logistic.py -------------------------------------------------------------------------------- /main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/main_pretrain.py -------------------------------------------------------------------------------- /models_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/models_sim.py -------------------------------------------------------------------------------- /models_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/models_vit.py -------------------------------------------------------------------------------- /util/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/augmentation.py -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/datasets.py -------------------------------------------------------------------------------- /util/lars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/lars.py -------------------------------------------------------------------------------- /util/lr_decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/lr_decay.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/masking_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/masking_generator.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/pos_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/pos_embed.py -------------------------------------------------------------------------------- /util/tcs_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/Siamese-Image-Modeling/HEAD/util/tcs_datasets.py --------------------------------------------------------------------------------