├── .gitignore ├── README.md ├── configs ├── imagelt │ ├── finetune │ │ ├── finetune_r50.py │ │ └── finetune_vit16.py │ └── pretrain │ │ ├── pretrain_r50.py │ │ └── pretrain_vit16.py ├── inat │ ├── finetune │ │ ├── inat_finetune_r50.py │ │ └── inat_finetune_vit16.py │ └── pretrain │ │ ├── inat_pretrain_r50.py │ │ └── inat_pretrain_vit16.py └── placeslt │ ├── finetune │ ├── places_finetune_r50.py │ └── places_finetune_vit16.py │ └── pretrain │ ├── places_pretrain_r50.py │ └── places_pretrain_vit16.py ├── datasets.py ├── dist_train_arun.sh ├── engine.py ├── eval.sh ├── hubconf.py ├── losses.py ├── main.py ├── mcloader ├── __init__.py ├── classification.py ├── image_list.py ├── imagenet.py ├── inat.py ├── preprocess.py └── prompt_template.py ├── mixup.py ├── models ├── __init__.py ├── finetune.py ├── layers │ ├── GatherLayer.py │ ├── ModulatedAttLayer.py │ ├── __init__.py │ ├── drop.py │ └── weight_init.py └── pretrain.py ├── optim_factory.py ├── requirements.txt ├── samplers.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/README.md -------------------------------------------------------------------------------- /configs/imagelt/finetune/finetune_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/imagelt/finetune/finetune_r50.py -------------------------------------------------------------------------------- /configs/imagelt/finetune/finetune_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/imagelt/finetune/finetune_vit16.py -------------------------------------------------------------------------------- /configs/imagelt/pretrain/pretrain_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/imagelt/pretrain/pretrain_r50.py -------------------------------------------------------------------------------- /configs/imagelt/pretrain/pretrain_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/imagelt/pretrain/pretrain_vit16.py -------------------------------------------------------------------------------- /configs/inat/finetune/inat_finetune_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/inat/finetune/inat_finetune_r50.py -------------------------------------------------------------------------------- /configs/inat/finetune/inat_finetune_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/inat/finetune/inat_finetune_vit16.py -------------------------------------------------------------------------------- /configs/inat/pretrain/inat_pretrain_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/inat/pretrain/inat_pretrain_r50.py -------------------------------------------------------------------------------- /configs/inat/pretrain/inat_pretrain_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/inat/pretrain/inat_pretrain_vit16.py -------------------------------------------------------------------------------- /configs/placeslt/finetune/places_finetune_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/placeslt/finetune/places_finetune_r50.py -------------------------------------------------------------------------------- /configs/placeslt/finetune/places_finetune_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/placeslt/finetune/places_finetune_vit16.py -------------------------------------------------------------------------------- /configs/placeslt/pretrain/places_pretrain_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/placeslt/pretrain/places_pretrain_r50.py -------------------------------------------------------------------------------- /configs/placeslt/pretrain/places_pretrain_vit16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/configs/placeslt/pretrain/places_pretrain_vit16.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/datasets.py -------------------------------------------------------------------------------- /dist_train_arun.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/dist_train_arun.sh -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/engine.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/eval.sh -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/hubconf.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/losses.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/main.py -------------------------------------------------------------------------------- /mcloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/__init__.py -------------------------------------------------------------------------------- /mcloader/classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/classification.py -------------------------------------------------------------------------------- /mcloader/image_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/image_list.py -------------------------------------------------------------------------------- /mcloader/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/imagenet.py -------------------------------------------------------------------------------- /mcloader/inat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/inat.py -------------------------------------------------------------------------------- /mcloader/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/preprocess.py -------------------------------------------------------------------------------- /mcloader/prompt_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mcloader/prompt_template.py -------------------------------------------------------------------------------- /mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/mixup.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/finetune.py -------------------------------------------------------------------------------- /models/layers/GatherLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/layers/GatherLayer.py -------------------------------------------------------------------------------- /models/layers/ModulatedAttLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/layers/ModulatedAttLayer.py -------------------------------------------------------------------------------- /models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/layers/__init__.py -------------------------------------------------------------------------------- /models/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/layers/drop.py -------------------------------------------------------------------------------- /models/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/layers/weight_init.py -------------------------------------------------------------------------------- /models/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/models/pretrain.py -------------------------------------------------------------------------------- /optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/optim_factory.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/requirements.txt -------------------------------------------------------------------------------- /samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/samplers.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChangyaoTian/VL-LTR/HEAD/utils.py --------------------------------------------------------------------------------