├── README.md ├── dali.py ├── data2rec.py ├── dataloaders ├── __init__.py ├── caltech101.py ├── cifar10.py ├── cifar100.py ├── coco.py ├── dtd.py ├── fgvc_aircraft.py ├── flickr30k.py ├── flowers102.py ├── food101.py ├── imagenet.py ├── oxford_pets.py ├── stanford_car.py ├── sun397.py └── utils.py ├── img └── architecture.jpg ├── label.json ├── requirments.txt ├── run_retrieval.sh ├── run_zeroshot.sh ├── scripts └── train_yfcc15m_B32_ALIP.sh ├── src └── open_alip │ ├── __init__.py │ ├── bpe_simple_vocab_16e6.txt.gz │ ├── constants.py │ ├── factory.py │ ├── loss.py │ ├── model.py │ ├── model_configs │ ├── RN101.json │ ├── RN50.json │ ├── ViT-B-16.json │ └── ViT-B-32.json │ ├── tokenizer.py │ ├── transform.py │ └── utils.py ├── templates.json ├── text_image_retrieval.py ├── train.py └── zero_shot.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/README.md -------------------------------------------------------------------------------- /dali.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dali.py -------------------------------------------------------------------------------- /data2rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/data2rec.py -------------------------------------------------------------------------------- /dataloaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataloaders/caltech101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/caltech101.py -------------------------------------------------------------------------------- /dataloaders/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/cifar10.py -------------------------------------------------------------------------------- /dataloaders/cifar100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/cifar100.py -------------------------------------------------------------------------------- /dataloaders/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/coco.py -------------------------------------------------------------------------------- /dataloaders/dtd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/dtd.py -------------------------------------------------------------------------------- /dataloaders/fgvc_aircraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/fgvc_aircraft.py -------------------------------------------------------------------------------- /dataloaders/flickr30k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/flickr30k.py -------------------------------------------------------------------------------- /dataloaders/flowers102.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/flowers102.py -------------------------------------------------------------------------------- /dataloaders/food101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/food101.py -------------------------------------------------------------------------------- /dataloaders/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/imagenet.py -------------------------------------------------------------------------------- /dataloaders/oxford_pets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/oxford_pets.py -------------------------------------------------------------------------------- /dataloaders/stanford_car.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/stanford_car.py -------------------------------------------------------------------------------- /dataloaders/sun397.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/sun397.py -------------------------------------------------------------------------------- /dataloaders/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/dataloaders/utils.py -------------------------------------------------------------------------------- /img/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/img/architecture.jpg -------------------------------------------------------------------------------- /label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/label.json -------------------------------------------------------------------------------- /requirments.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/requirments.txt -------------------------------------------------------------------------------- /run_retrieval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/run_retrieval.sh -------------------------------------------------------------------------------- /run_zeroshot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/run_zeroshot.sh -------------------------------------------------------------------------------- /scripts/train_yfcc15m_B32_ALIP.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/scripts/train_yfcc15m_B32_ALIP.sh -------------------------------------------------------------------------------- /src/open_alip/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/__init__.py -------------------------------------------------------------------------------- /src/open_alip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /src/open_alip/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/constants.py -------------------------------------------------------------------------------- /src/open_alip/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/factory.py -------------------------------------------------------------------------------- /src/open_alip/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/loss.py -------------------------------------------------------------------------------- /src/open_alip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/model.py -------------------------------------------------------------------------------- /src/open_alip/model_configs/RN101.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/model_configs/RN101.json -------------------------------------------------------------------------------- /src/open_alip/model_configs/RN50.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/model_configs/RN50.json -------------------------------------------------------------------------------- /src/open_alip/model_configs/ViT-B-16.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/model_configs/ViT-B-16.json -------------------------------------------------------------------------------- /src/open_alip/model_configs/ViT-B-32.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/model_configs/ViT-B-32.json -------------------------------------------------------------------------------- /src/open_alip/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/tokenizer.py -------------------------------------------------------------------------------- /src/open_alip/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/transform.py -------------------------------------------------------------------------------- /src/open_alip/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/src/open_alip/utils.py -------------------------------------------------------------------------------- /templates.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/templates.json -------------------------------------------------------------------------------- /text_image_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/text_image_retrieval.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/train.py -------------------------------------------------------------------------------- /zero_shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepglint/ALIP/HEAD/zero_shot.py --------------------------------------------------------------------------------