├── .gitignore ├── LICENSE ├── README.md ├── figs └── probvlm.png ├── requirements.txt └── src ├── clip ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-37.pyc │ ├── clip.cpython-310.pyc │ ├── clip.cpython-37.pyc │ ├── model.cpython-310.pyc │ ├── model.cpython-37.pyc │ ├── simple_tokenizer.cpython-310.pyc │ └── simple_tokenizer.cpython-37.pyc ├── clip.py ├── model.py └── simple_tokenizer.py ├── ds ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── __init__.cpython-37.pyc │ ├── _dataloader.cpython-310.pyc │ ├── _dataloader.cpython-37.pyc │ ├── _transforms.cpython-310.pyc │ ├── _transforms.cpython-37.pyc │ ├── coco.cpython-310.pyc │ ├── coco.cpython-37.pyc │ ├── cub.cpython-310.pyc │ ├── cub.cpython-37.pyc │ ├── fashion200k.cpython-310.pyc │ ├── fashion200k.cpython-37.pyc │ ├── flickr.cpython-310.pyc │ ├── flickr.cpython-37.pyc │ ├── flo.cpython-310.pyc │ ├── flo.cpython-37.pyc │ ├── simple_tokenizer.cpython-310.pyc │ ├── simple_tokenizer.cpython-37.pyc │ ├── vocab.cpython-310.pyc │ └── vocab.cpython-37.pyc ├── _dataloader.py ├── _transforms.py ├── annotations │ ├── cub │ │ ├── seen_test_images.txt │ │ ├── testclasses.txt │ │ ├── trainclasses1.txt │ │ ├── trainclasses2.txt │ │ ├── trainclasses3.txt │ │ ├── trainvalclasses.txt │ │ ├── valclasses1.txt │ │ ├── valclasses2.txt │ │ └── valclasses3.txt │ ├── flickr │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt │ └── flo │ │ ├── testclasses.txt │ │ └── trainvalclasses.txt ├── coco.py ├── cub.py ├── fashion200k.py ├── flickr.py ├── flo.py ├── simple_tokenizer.py ├── vocab.py └── vocabs │ └── make_vocab.py ├── ds_lavis ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── _dataloader.cpython-310.pyc │ ├── coco.cpython-310.pyc │ ├── cub.cpython-310.pyc │ ├── fashion200k.cpython-310.pyc │ ├── flickr.cpython-310.pyc │ ├── flo.cpython-310.pyc │ ├── simple_tokenizer.cpython-310.pyc │ └── vocab.cpython-310.pyc ├── _dataloader.py ├── _transforms.py ├── annotations │ ├── cub │ │ ├── seen_test_images.txt │ │ ├── testclasses.txt │ │ ├── trainclasses1.txt │ │ ├── trainclasses2.txt │ │ ├── trainclasses3.txt │ │ ├── trainvalclasses.txt │ │ ├── valclasses1.txt │ │ ├── valclasses2.txt │ │ └── valclasses3.txt │ ├── flickr │ │ ├── test.txt │ │ ├── train.txt │ │ └── val.txt │ └── flo │ │ ├── testclasses.txt │ │ └── trainvalclasses.txt ├── coco.py ├── cub.py ├── fashion200k.py ├── flickr.py ├── flo.py ├── simple_tokenizer.py ├── vocab.py └── vocabs │ └── make_vocab.py ├── losses.py ├── networks.py ├── train_ProbVLM_CLIP.ipynb ├── train_probVLM.py ├── utils.py └── utils_lavis.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/README.md -------------------------------------------------------------------------------- /figs/probvlm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/figs/probvlm.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /src/clip/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/clip.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/clip.cpython-310.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/clip.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/clip.cpython-37.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/model.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/model.cpython-310.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/model.cpython-37.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/simple_tokenizer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/simple_tokenizer.cpython-310.pyc -------------------------------------------------------------------------------- /src/clip/__pycache__/simple_tokenizer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/__pycache__/simple_tokenizer.cpython-37.pyc -------------------------------------------------------------------------------- /src/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/clip.py -------------------------------------------------------------------------------- /src/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/model.py -------------------------------------------------------------------------------- /src/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /src/ds/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__init__.py -------------------------------------------------------------------------------- /src/ds/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/_dataloader.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/_dataloader.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/_dataloader.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/_dataloader.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/_transforms.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/_transforms.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/_transforms.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/_transforms.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/coco.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/coco.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/coco.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/coco.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/cub.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/cub.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/cub.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/cub.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/fashion200k.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/fashion200k.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/fashion200k.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/fashion200k.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/flickr.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/flickr.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/flickr.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/flickr.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/flo.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/flo.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/flo.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/flo.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/simple_tokenizer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/simple_tokenizer.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/simple_tokenizer.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/simple_tokenizer.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/vocab.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/vocab.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds/__pycache__/vocab.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/__pycache__/vocab.cpython-37.pyc -------------------------------------------------------------------------------- /src/ds/_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/_dataloader.py -------------------------------------------------------------------------------- /src/ds/_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/_transforms.py -------------------------------------------------------------------------------- /src/ds/annotations/cub/seen_test_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/seen_test_images.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/testclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/testclasses.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/trainclasses1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/trainclasses1.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/trainclasses2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/trainclasses2.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/trainclasses3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/trainclasses3.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/trainvalclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/trainvalclasses.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/valclasses1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/valclasses1.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/valclasses2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/valclasses2.txt -------------------------------------------------------------------------------- /src/ds/annotations/cub/valclasses3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/cub/valclasses3.txt -------------------------------------------------------------------------------- /src/ds/annotations/flickr/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/flickr/test.txt -------------------------------------------------------------------------------- /src/ds/annotations/flickr/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/flickr/train.txt -------------------------------------------------------------------------------- /src/ds/annotations/flickr/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/flickr/val.txt -------------------------------------------------------------------------------- /src/ds/annotations/flo/testclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/flo/testclasses.txt -------------------------------------------------------------------------------- /src/ds/annotations/flo/trainvalclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/annotations/flo/trainvalclasses.txt -------------------------------------------------------------------------------- /src/ds/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/coco.py -------------------------------------------------------------------------------- /src/ds/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/cub.py -------------------------------------------------------------------------------- /src/ds/fashion200k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/fashion200k.py -------------------------------------------------------------------------------- /src/ds/flickr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/flickr.py -------------------------------------------------------------------------------- /src/ds/flo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/flo.py -------------------------------------------------------------------------------- /src/ds/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/simple_tokenizer.py -------------------------------------------------------------------------------- /src/ds/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/vocab.py -------------------------------------------------------------------------------- /src/ds/vocabs/make_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds/vocabs/make_vocab.py -------------------------------------------------------------------------------- /src/ds_lavis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__init__.py -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/_dataloader.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/_dataloader.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/coco.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/coco.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/cub.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/cub.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/fashion200k.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/fashion200k.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/flickr.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/flickr.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/flo.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/flo.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/simple_tokenizer.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/simple_tokenizer.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/__pycache__/vocab.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/__pycache__/vocab.cpython-310.pyc -------------------------------------------------------------------------------- /src/ds_lavis/_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/_dataloader.py -------------------------------------------------------------------------------- /src/ds_lavis/_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/_transforms.py -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/seen_test_images.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/seen_test_images.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/testclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/testclasses.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/trainclasses1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/trainclasses1.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/trainclasses2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/trainclasses2.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/trainclasses3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/trainclasses3.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/trainvalclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/trainvalclasses.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/valclasses1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/valclasses1.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/valclasses2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/valclasses2.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/cub/valclasses3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/cub/valclasses3.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/flickr/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/flickr/test.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/flickr/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/flickr/train.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/flickr/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/flickr/val.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/flo/testclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/flo/testclasses.txt -------------------------------------------------------------------------------- /src/ds_lavis/annotations/flo/trainvalclasses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/annotations/flo/trainvalclasses.txt -------------------------------------------------------------------------------- /src/ds_lavis/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/coco.py -------------------------------------------------------------------------------- /src/ds_lavis/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/cub.py -------------------------------------------------------------------------------- /src/ds_lavis/fashion200k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/fashion200k.py -------------------------------------------------------------------------------- /src/ds_lavis/flickr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/flickr.py -------------------------------------------------------------------------------- /src/ds_lavis/flo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/flo.py -------------------------------------------------------------------------------- /src/ds_lavis/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/simple_tokenizer.py -------------------------------------------------------------------------------- /src/ds_lavis/vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/vocab.py -------------------------------------------------------------------------------- /src/ds_lavis/vocabs/make_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/ds_lavis/vocabs/make_vocab.py -------------------------------------------------------------------------------- /src/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/losses.py -------------------------------------------------------------------------------- /src/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/networks.py -------------------------------------------------------------------------------- /src/train_ProbVLM_CLIP.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/train_ProbVLM_CLIP.ipynb -------------------------------------------------------------------------------- /src/train_probVLM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/train_probVLM.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/utils.py -------------------------------------------------------------------------------- /src/utils_lavis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ExplainableML/ProbVLM/HEAD/src/utils_lavis.py --------------------------------------------------------------------------------