├── .gitignore ├── README.md ├── convert_to_tfrecord_from_file.py ├── download_models.sh ├── prepare_data.sh └── universal_embedding ├── app.py ├── classification_model.py ├── classification_with_knn_eval_trainer.py ├── clip_vit_with_embedding.py ├── configs ├── config_descr_eval.py ├── config_extract_dir_descriptors.py ├── config_knn_vit.py └── config_train_vit.py ├── dataset_infos.py ├── datasets.py ├── descr_eval.py ├── extract_dir_descriptors.py ├── iccv_configs ├── CLIP_vitb_ujcds_train.py ├── CLIP_vitb_ujcrr_train.py ├── CLIP_vitb_uscds_train.py ├── CLIP_vitb_uscrr_train.py ├── IN_vitb_ujcds_train.py ├── IN_vitb_ujcrr_train.py ├── IN_vitb_uscds_train.py └── IN_vitb_uscrr_train.py ├── info_utils.py ├── knn_main.py ├── knn_utils.py ├── loss_utils.py ├── main.py ├── model_init.py ├── models.py ├── optimizer_utils.py ├── sampling_utils.py ├── train_eval_steps.py ├── utils.py └── vit_with_embedding.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | data 3 | wandb 4 | results 5 | scripts 6 | logs -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/README.md -------------------------------------------------------------------------------- /convert_to_tfrecord_from_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/convert_to_tfrecord_from_file.py -------------------------------------------------------------------------------- /download_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/download_models.sh -------------------------------------------------------------------------------- /prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/prepare_data.sh -------------------------------------------------------------------------------- /universal_embedding/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/app.py -------------------------------------------------------------------------------- /universal_embedding/classification_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/classification_model.py -------------------------------------------------------------------------------- /universal_embedding/classification_with_knn_eval_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/classification_with_knn_eval_trainer.py -------------------------------------------------------------------------------- /universal_embedding/clip_vit_with_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/clip_vit_with_embedding.py -------------------------------------------------------------------------------- /universal_embedding/configs/config_descr_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/configs/config_descr_eval.py -------------------------------------------------------------------------------- /universal_embedding/configs/config_extract_dir_descriptors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/configs/config_extract_dir_descriptors.py -------------------------------------------------------------------------------- /universal_embedding/configs/config_knn_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/configs/config_knn_vit.py -------------------------------------------------------------------------------- /universal_embedding/configs/config_train_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/configs/config_train_vit.py -------------------------------------------------------------------------------- /universal_embedding/dataset_infos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/dataset_infos.py -------------------------------------------------------------------------------- /universal_embedding/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/datasets.py -------------------------------------------------------------------------------- /universal_embedding/descr_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/descr_eval.py -------------------------------------------------------------------------------- /universal_embedding/extract_dir_descriptors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/extract_dir_descriptors.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/CLIP_vitb_ujcds_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/CLIP_vitb_ujcds_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/CLIP_vitb_ujcrr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/CLIP_vitb_ujcrr_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/CLIP_vitb_uscds_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/CLIP_vitb_uscds_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/CLIP_vitb_uscrr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/CLIP_vitb_uscrr_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/IN_vitb_ujcds_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/IN_vitb_ujcds_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/IN_vitb_ujcrr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/IN_vitb_ujcrr_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/IN_vitb_uscds_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/IN_vitb_uscds_train.py -------------------------------------------------------------------------------- /universal_embedding/iccv_configs/IN_vitb_uscrr_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/iccv_configs/IN_vitb_uscrr_train.py -------------------------------------------------------------------------------- /universal_embedding/info_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/info_utils.py -------------------------------------------------------------------------------- /universal_embedding/knn_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/knn_main.py -------------------------------------------------------------------------------- /universal_embedding/knn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/knn_utils.py -------------------------------------------------------------------------------- /universal_embedding/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/loss_utils.py -------------------------------------------------------------------------------- /universal_embedding/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/main.py -------------------------------------------------------------------------------- /universal_embedding/model_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/model_init.py -------------------------------------------------------------------------------- /universal_embedding/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/models.py -------------------------------------------------------------------------------- /universal_embedding/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/optimizer_utils.py -------------------------------------------------------------------------------- /universal_embedding/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/sampling_utils.py -------------------------------------------------------------------------------- /universal_embedding/train_eval_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/train_eval_steps.py -------------------------------------------------------------------------------- /universal_embedding/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/utils.py -------------------------------------------------------------------------------- /universal_embedding/vit_with_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nikosips/Universal-Image-Embeddings/HEAD/universal_embedding/vit_with_embedding.py --------------------------------------------------------------------------------