├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── configs ├── data │ ├── cirr.yaml │ ├── fashioniq-base.yaml │ ├── fashioniq-dress.yaml │ ├── fashioniq-shirt.yaml │ ├── fashioniq-toptee.yaml │ ├── fashioniq.yaml │ ├── webvid-covr.yaml │ └── webvid-covr_rule-based.yaml ├── experiment │ ├── cirr.yaml │ ├── covr_hard-negatives.yaml │ ├── covr_iterate-triplets.yaml │ ├── covr_late-fusion.yaml │ ├── covr_middle-emb.yaml │ ├── covr_only-text.yaml │ ├── covr_only-visual.yaml │ ├── covr_random-frame.yaml │ ├── covr_rule-based.yaml │ └── fiq-all.yaml ├── machine │ ├── default.yaml │ └── server.yaml ├── med_config.json ├── model │ ├── blip-large.yaml │ ├── blip-large_text.yaml │ ├── blip-large_visual.yaml │ ├── ckpt │ │ ├── blip-l-coco.yaml │ │ ├── cirr-gt.yaml │ │ ├── cirr_ft-covr+gt.yaml │ │ ├── fashioniq-all-ft_covr.yaml │ │ ├── webvid-covr-test.yaml │ │ └── webvid-covr.yaml │ ├── loss │ │ ├── cross_entropy.yaml │ │ └── hn_nce.yaml │ ├── optimizer │ │ └── adamw.yaml │ └── scheduler │ │ ├── cosine.yaml │ │ └── step.yaml ├── test.yaml ├── test │ ├── all.yaml │ ├── cirr.yaml │ ├── fashioniq-dress.yaml │ ├── fashioniq-shirt.yaml │ ├── fashioniq-toptee.yaml │ ├── fashioniq.yaml │ ├── main.yaml │ ├── webvid-covr.yaml │ ├── webvid-covr_text.yaml │ └── webvid-covr_visual.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── logger │ ├── aim.yaml │ ├── comet.yaml │ ├── csv.yaml │ ├── mlflow.yaml │ ├── neptune.yaml │ ├── none.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── requirement.txt ├── slurm_test.sh ├── slurm_train.sh ├── src ├── data │ ├── cirr.py │ ├── embs.py │ ├── fashioniq.py │ ├── randaugment.py │ ├── transforms.py │ ├── utils.py │ ├── webvid_covr.py │ └── webvid_covr_rulebased.py ├── model │ ├── blip.py │ ├── blip_cir.py │ ├── blip_cir_text.py │ ├── blip_cir_visual.py │ ├── blip_embs.py │ ├── loss.py │ ├── med.py │ └── vit.py ├── test │ ├── cirr.py │ ├── fashioniq.py │ ├── utils.py │ ├── webvid_covr.py │ └── webvid_covr_exp.py └── tools │ ├── files.py │ ├── scheduler.py │ ├── utils.py │ └── visualize.py ├── test.py ├── tools ├── embs │ ├── save_blip_embs_imgs.py │ ├── save_blip_embs_vids.py │ └── save_blip_embs_vids_multimodal.py ├── examples │ └── teaser.gif ├── notebooks │ └── visualize_dataset.ipynb └── scripts │ ├── download_annotation.sh │ ├── download_annotation_cirr.sh │ ├── download_annotation_covr.sh │ ├── download_annotation_fiq.sh │ ├── download_covr.py │ ├── download_pretrained_models.sh │ ├── merge_fiq_annotations.py │ ├── webvid_descriptions_2m.py │ └── webvid_descriptions_8m.py └── train.py /.gitattributes: -------------------------------------------------------------------------------- 1 | *.ipynb linguist-vendored -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/README.md -------------------------------------------------------------------------------- /configs/data/cirr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/data/cirr.yaml -------------------------------------------------------------------------------- /configs/data/fashioniq-base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/data/fashioniq-base.yaml -------------------------------------------------------------------------------- /configs/data/fashioniq-dress.yaml: -------------------------------------------------------------------------------- 1 | defaults: 2 | - fashioniq-base.yaml 3 | 4 | category: dress 5 | -------------------------------------------------------------------------------- /configs/data/fashioniq-shirt.yaml: -------------------------------------------------------------------------------- 1 | defaults: 2 | - fashioniq-base.yaml 3 | 4 | category: shirt 5 | -------------------------------------------------------------------------------- /configs/data/fashioniq-toptee.yaml: -------------------------------------------------------------------------------- 1 | defaults: 2 | - fashioniq-base.yaml 3 | 4 | category: toptee 5 | -------------------------------------------------------------------------------- /configs/data/fashioniq.yaml: -------------------------------------------------------------------------------- 1 | defaults: 2 | - fashioniq-base.yaml 3 | 4 | category: all 5 | -------------------------------------------------------------------------------- /configs/data/webvid-covr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/data/webvid-covr.yaml -------------------------------------------------------------------------------- /configs/data/webvid-covr_rule-based.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/data/webvid-covr_rule-based.yaml -------------------------------------------------------------------------------- /configs/experiment/cirr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/cirr.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_hard-negatives.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_hard-negatives.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_iterate-triplets.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_iterate-triplets.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_late-fusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_late-fusion.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_middle-emb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_middle-emb.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_only-text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_only-text.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_only-visual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_only-visual.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_random-frame.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_random-frame.yaml -------------------------------------------------------------------------------- /configs/experiment/covr_rule-based.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/covr_rule-based.yaml -------------------------------------------------------------------------------- /configs/experiment/fiq-all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/experiment/fiq-all.yaml -------------------------------------------------------------------------------- /configs/machine/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/machine/default.yaml -------------------------------------------------------------------------------- /configs/machine/server.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/machine/server.yaml -------------------------------------------------------------------------------- /configs/med_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/med_config.json -------------------------------------------------------------------------------- /configs/model/blip-large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/blip-large.yaml -------------------------------------------------------------------------------- /configs/model/blip-large_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/blip-large_text.yaml -------------------------------------------------------------------------------- /configs/model/blip-large_visual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/blip-large_visual.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/blip-l-coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/blip-l-coco.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/cirr-gt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/cirr-gt.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/cirr_ft-covr+gt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/cirr_ft-covr+gt.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/fashioniq-all-ft_covr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/fashioniq-all-ft_covr.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/webvid-covr-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/webvid-covr-test.yaml -------------------------------------------------------------------------------- /configs/model/ckpt/webvid-covr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/ckpt/webvid-covr.yaml -------------------------------------------------------------------------------- /configs/model/loss/cross_entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/loss/cross_entropy.yaml -------------------------------------------------------------------------------- /configs/model/loss/hn_nce.yaml: -------------------------------------------------------------------------------- 1 | _target_: src.model.loss.HardNegativeNCE 2 | name: hnnce 3 | 4 | alpha: 1 5 | beta: 0.5 6 | -------------------------------------------------------------------------------- /configs/model/optimizer/adamw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/optimizer/adamw.yaml -------------------------------------------------------------------------------- /configs/model/scheduler/cosine.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/scheduler/cosine.yaml -------------------------------------------------------------------------------- /configs/model/scheduler/step.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/model/scheduler/step.yaml -------------------------------------------------------------------------------- /configs/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test.yaml -------------------------------------------------------------------------------- /configs/test/all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/all.yaml -------------------------------------------------------------------------------- /configs/test/cirr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/cirr.yaml -------------------------------------------------------------------------------- /configs/test/fashioniq-dress.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/fashioniq-dress.yaml -------------------------------------------------------------------------------- /configs/test/fashioniq-shirt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/fashioniq-shirt.yaml -------------------------------------------------------------------------------- /configs/test/fashioniq-toptee.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/fashioniq-toptee.yaml -------------------------------------------------------------------------------- /configs/test/fashioniq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/fashioniq.yaml -------------------------------------------------------------------------------- /configs/test/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/main.yaml -------------------------------------------------------------------------------- /configs/test/webvid-covr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/webvid-covr.yaml -------------------------------------------------------------------------------- /configs/test/webvid-covr_text.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/webvid-covr_text.yaml -------------------------------------------------------------------------------- /configs/test/webvid-covr_visual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/test/webvid-covr_visual.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/aim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/aim.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/comet.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/csv.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/none.yaml: -------------------------------------------------------------------------------- 1 | logger_name: none 2 | 3 | project: null 4 | -------------------------------------------------------------------------------- /configs/trainer/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/trainer/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/configs/trainer/logger/wandb.yaml -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/requirement.txt -------------------------------------------------------------------------------- /slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/slurm_test.sh -------------------------------------------------------------------------------- /slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/slurm_train.sh -------------------------------------------------------------------------------- /src/data/cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/cirr.py -------------------------------------------------------------------------------- /src/data/embs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/embs.py -------------------------------------------------------------------------------- /src/data/fashioniq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/fashioniq.py -------------------------------------------------------------------------------- /src/data/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/randaugment.py -------------------------------------------------------------------------------- /src/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/transforms.py -------------------------------------------------------------------------------- /src/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/utils.py -------------------------------------------------------------------------------- /src/data/webvid_covr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/webvid_covr.py -------------------------------------------------------------------------------- /src/data/webvid_covr_rulebased.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/data/webvid_covr_rulebased.py -------------------------------------------------------------------------------- /src/model/blip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/blip.py -------------------------------------------------------------------------------- /src/model/blip_cir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/blip_cir.py -------------------------------------------------------------------------------- /src/model/blip_cir_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/blip_cir_text.py -------------------------------------------------------------------------------- /src/model/blip_cir_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/blip_cir_visual.py -------------------------------------------------------------------------------- /src/model/blip_embs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/blip_embs.py -------------------------------------------------------------------------------- /src/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/loss.py -------------------------------------------------------------------------------- /src/model/med.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/med.py -------------------------------------------------------------------------------- /src/model/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/model/vit.py -------------------------------------------------------------------------------- /src/test/cirr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/test/cirr.py -------------------------------------------------------------------------------- /src/test/fashioniq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/test/fashioniq.py -------------------------------------------------------------------------------- /src/test/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/test/utils.py -------------------------------------------------------------------------------- /src/test/webvid_covr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/test/webvid_covr.py -------------------------------------------------------------------------------- /src/test/webvid_covr_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/test/webvid_covr_exp.py -------------------------------------------------------------------------------- /src/tools/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/tools/files.py -------------------------------------------------------------------------------- /src/tools/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/tools/scheduler.py -------------------------------------------------------------------------------- /src/tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/tools/utils.py -------------------------------------------------------------------------------- /src/tools/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/src/tools/visualize.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/test.py -------------------------------------------------------------------------------- /tools/embs/save_blip_embs_imgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/embs/save_blip_embs_imgs.py -------------------------------------------------------------------------------- /tools/embs/save_blip_embs_vids.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/embs/save_blip_embs_vids.py -------------------------------------------------------------------------------- /tools/embs/save_blip_embs_vids_multimodal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/embs/save_blip_embs_vids_multimodal.py -------------------------------------------------------------------------------- /tools/examples/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/examples/teaser.gif -------------------------------------------------------------------------------- /tools/notebooks/visualize_dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/notebooks/visualize_dataset.ipynb -------------------------------------------------------------------------------- /tools/scripts/download_annotation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_annotation.sh -------------------------------------------------------------------------------- /tools/scripts/download_annotation_cirr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_annotation_cirr.sh -------------------------------------------------------------------------------- /tools/scripts/download_annotation_covr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_annotation_covr.sh -------------------------------------------------------------------------------- /tools/scripts/download_annotation_fiq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_annotation_fiq.sh -------------------------------------------------------------------------------- /tools/scripts/download_covr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_covr.py -------------------------------------------------------------------------------- /tools/scripts/download_pretrained_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/download_pretrained_models.sh -------------------------------------------------------------------------------- /tools/scripts/merge_fiq_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/merge_fiq_annotations.py -------------------------------------------------------------------------------- /tools/scripts/webvid_descriptions_2m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/webvid_descriptions_2m.py -------------------------------------------------------------------------------- /tools/scripts/webvid_descriptions_8m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/tools/scripts/webvid_descriptions_8m.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OmkarThawakar/composed-video-retrieval/HEAD/train.py --------------------------------------------------------------------------------