├── .gitignore ├── LICENSE ├── RAdam.py ├── README.md ├── combine_sampler.py ├── config ├── config_cars_test.yaml ├── config_cars_test_pseudo.yaml ├── config_cars_train.yaml ├── config_cub_test.yaml ├── config_cub_test_pseudo.yaml ├── config_cub_train.yaml ├── config_inshop_test.yaml ├── config_inshop_train.yaml ├── config_sop_test.yaml └── config_sop_train.yaml ├── data_utility.py ├── dataset ├── __init__.py ├── cub.py ├── inshop.py ├── parser_SOP.py ├── parser_in_shop.py └── utils.py ├── environment.yaml ├── evaluation ├── __init__.py ├── normalized_mutual_information.py ├── recall.py ├── recall_inshop.py └── utils.py ├── net ├── __init__.py ├── attentions.py ├── densenet.py ├── embed.py ├── gnn_base.py ├── graph_generator.py ├── inception_bn.py ├── load_trained_net.py ├── resnet.py └── utils.py ├── storage └── search_results │ └── test_0.7029034436191762_cub_1623134434.931427 │ └── test_0.7029034436191762_cub_1623134434.931427.txt ├── train.py ├── trainer.py └── utils ├── losses.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/LICENSE -------------------------------------------------------------------------------- /RAdam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/RAdam.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/README.md -------------------------------------------------------------------------------- /combine_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/combine_sampler.py -------------------------------------------------------------------------------- /config/config_cars_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cars_test.yaml -------------------------------------------------------------------------------- /config/config_cars_test_pseudo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cars_test_pseudo.yaml -------------------------------------------------------------------------------- /config/config_cars_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cars_train.yaml -------------------------------------------------------------------------------- /config/config_cub_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cub_test.yaml -------------------------------------------------------------------------------- /config/config_cub_test_pseudo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cub_test_pseudo.yaml -------------------------------------------------------------------------------- /config/config_cub_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_cub_train.yaml -------------------------------------------------------------------------------- /config/config_inshop_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_inshop_test.yaml -------------------------------------------------------------------------------- /config/config_inshop_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_inshop_train.yaml -------------------------------------------------------------------------------- /config/config_sop_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_sop_test.yaml -------------------------------------------------------------------------------- /config/config_sop_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/config/config_sop_train.yaml -------------------------------------------------------------------------------- /data_utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/data_utility.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/cub.py -------------------------------------------------------------------------------- /dataset/inshop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/inshop.py -------------------------------------------------------------------------------- /dataset/parser_SOP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/parser_SOP.py -------------------------------------------------------------------------------- /dataset/parser_in_shop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/parser_in_shop.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/environment.yaml -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/normalized_mutual_information.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/evaluation/normalized_mutual_information.py -------------------------------------------------------------------------------- /evaluation/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/evaluation/recall.py -------------------------------------------------------------------------------- /evaluation/recall_inshop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/evaluation/recall_inshop.py -------------------------------------------------------------------------------- /evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/evaluation/utils.py -------------------------------------------------------------------------------- /net/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/__init__.py -------------------------------------------------------------------------------- /net/attentions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/attentions.py -------------------------------------------------------------------------------- /net/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/densenet.py -------------------------------------------------------------------------------- /net/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/embed.py -------------------------------------------------------------------------------- /net/gnn_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/gnn_base.py -------------------------------------------------------------------------------- /net/graph_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/graph_generator.py -------------------------------------------------------------------------------- /net/inception_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/inception_bn.py -------------------------------------------------------------------------------- /net/load_trained_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/load_trained_net.py -------------------------------------------------------------------------------- /net/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/resnet.py -------------------------------------------------------------------------------- /net/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/net/utils.py -------------------------------------------------------------------------------- /storage/search_results/test_0.7029034436191762_cub_1623134434.931427/test_0.7029034436191762_cub_1623134434.931427.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/storage/search_results/test_0.7029034436191762_cub_1623134434.931427/test_0.7029034436191762_cub_1623134434.931427.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvl-tum/intra_batch/HEAD/utils/utils.py --------------------------------------------------------------------------------