├── .gitmodules ├── README.md ├── checkpoints ├── fg-ovd_linear-mlp.pt ├── fg-ovd_linear-mlp_te-freezed.pt ├── fg-ovd_linear-mlp_ve-freezed.pt ├── fg-ovd_non-linear-mlp.pt ├── triplet_linear-mlp.pt ├── triplet_linear-mlp_te-freezed.pt ├── triplet_linear-mlp_ve-freezed.pt └── triplet_non-linear-mlp.pt ├── configs ├── model │ ├── attention.yaml │ ├── linear-mlp.yaml │ ├── linear-mlp_te-freezed.yaml │ ├── linear-mlp_ve-freezed.yaml │ └── non-linear-mlp.yaml └── train │ ├── fg-ovd.yaml │ └── triplet.yaml ├── eval ├── README.md ├── paper_plots.ipynb └── test.py ├── features ├── coco_feature_extractor.py ├── create_karpathy_splits.py └── extract.py ├── fg-ovd_feature_extraction └── extraction.py ├── fgovd_evaluation ├── inference_on_benchmark.py └── main.py ├── plots.ipynb ├── requirements.txt ├── src ├── dataset.py ├── eval_util.py ├── loss.py ├── metrics.py ├── model.py ├── plots_util.py ├── ranks.py ├── train_util.py └── utilities.py └── train.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/fg-ovd_linear-mlp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/fg-ovd_linear-mlp.pt -------------------------------------------------------------------------------- /checkpoints/fg-ovd_linear-mlp_te-freezed.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/fg-ovd_linear-mlp_te-freezed.pt -------------------------------------------------------------------------------- /checkpoints/fg-ovd_linear-mlp_ve-freezed.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/fg-ovd_linear-mlp_ve-freezed.pt -------------------------------------------------------------------------------- /checkpoints/fg-ovd_non-linear-mlp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/fg-ovd_non-linear-mlp.pt -------------------------------------------------------------------------------- /checkpoints/triplet_linear-mlp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/triplet_linear-mlp.pt -------------------------------------------------------------------------------- /checkpoints/triplet_linear-mlp_te-freezed.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/triplet_linear-mlp_te-freezed.pt -------------------------------------------------------------------------------- /checkpoints/triplet_linear-mlp_ve-freezed.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/triplet_linear-mlp_ve-freezed.pt -------------------------------------------------------------------------------- /checkpoints/triplet_non-linear-mlp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/checkpoints/triplet_non-linear-mlp.pt -------------------------------------------------------------------------------- /configs/model/attention.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/model/attention.yaml -------------------------------------------------------------------------------- /configs/model/linear-mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/model/linear-mlp.yaml -------------------------------------------------------------------------------- /configs/model/linear-mlp_te-freezed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/model/linear-mlp_te-freezed.yaml -------------------------------------------------------------------------------- /configs/model/linear-mlp_ve-freezed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/model/linear-mlp_ve-freezed.yaml -------------------------------------------------------------------------------- /configs/model/non-linear-mlp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/model/non-linear-mlp.yaml -------------------------------------------------------------------------------- /configs/train/fg-ovd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/train/fg-ovd.yaml -------------------------------------------------------------------------------- /configs/train/triplet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/configs/train/triplet.yaml -------------------------------------------------------------------------------- /eval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/eval/README.md -------------------------------------------------------------------------------- /eval/paper_plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/eval/paper_plots.ipynb -------------------------------------------------------------------------------- /eval/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/eval/test.py -------------------------------------------------------------------------------- /features/coco_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/features/coco_feature_extractor.py -------------------------------------------------------------------------------- /features/create_karpathy_splits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/features/create_karpathy_splits.py -------------------------------------------------------------------------------- /features/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/features/extract.py -------------------------------------------------------------------------------- /fg-ovd_feature_extraction/extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/fg-ovd_feature_extraction/extraction.py -------------------------------------------------------------------------------- /fgovd_evaluation/inference_on_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/fgovd_evaluation/inference_on_benchmark.py -------------------------------------------------------------------------------- /fgovd_evaluation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/fgovd_evaluation/main.py -------------------------------------------------------------------------------- /plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/plots.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pyyaml 2 | matplotlib 3 | pandas 4 | seaborn 5 | tqdm -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/eval_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/eval_util.py -------------------------------------------------------------------------------- /src/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/loss.py -------------------------------------------------------------------------------- /src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/metrics.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/model.py -------------------------------------------------------------------------------- /src/plots_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/plots_util.py -------------------------------------------------------------------------------- /src/ranks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/ranks.py -------------------------------------------------------------------------------- /src/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/train_util.py -------------------------------------------------------------------------------- /src/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/src/utilities.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lorebianchi98/FG-CLIP/HEAD/train.py --------------------------------------------------------------------------------