├── LICENSE ├── README.md ├── conf ├── dataset │ ├── gldv2-test.yaml │ ├── gldv2-val.yaml │ ├── gldv2.yaml │ ├── roxford+1m.yaml │ ├── roxford.yaml │ ├── rparis+1m.yaml │ ├── rparis.yaml │ └── test_dataset.yaml ├── descriptors │ ├── cvnet.yaml │ └── dinov2.yaml ├── model │ └── ames.yaml ├── test.yaml └── train.yaml ├── extract ├── __init__.py ├── extract_cvnet.py ├── extract_descriptors.py ├── extract_dino.py ├── image_dataset.py ├── merge_hdf5.py ├── prepare_topk_global.py ├── spatial_attention_2d.py └── transforms.py ├── hubconf.py ├── requirements.txt └── src ├── __init__.py ├── evaluate.py ├── models ├── __init__.py ├── ames.py ├── binarization_layer.py ├── resnet.py └── transformer_dec.py ├── train.py └── utils ├── __init__.py ├── dataset_loader.py ├── metrics.py ├── revisited.py ├── tensor_dataset.py ├── training.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/README.md -------------------------------------------------------------------------------- /conf/dataset/gldv2-test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/gldv2-test.yaml -------------------------------------------------------------------------------- /conf/dataset/gldv2-val.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/gldv2-val.yaml -------------------------------------------------------------------------------- /conf/dataset/gldv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/gldv2.yaml -------------------------------------------------------------------------------- /conf/dataset/roxford+1m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/roxford+1m.yaml -------------------------------------------------------------------------------- /conf/dataset/roxford.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/roxford.yaml -------------------------------------------------------------------------------- /conf/dataset/rparis+1m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/rparis+1m.yaml -------------------------------------------------------------------------------- /conf/dataset/rparis.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/rparis.yaml -------------------------------------------------------------------------------- /conf/dataset/test_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/dataset/test_dataset.yaml -------------------------------------------------------------------------------- /conf/descriptors/cvnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/descriptors/cvnet.yaml -------------------------------------------------------------------------------- /conf/descriptors/dinov2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/descriptors/dinov2.yaml -------------------------------------------------------------------------------- /conf/model/ames.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/model/ames.yaml -------------------------------------------------------------------------------- /conf/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/test.yaml -------------------------------------------------------------------------------- /conf/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/conf/train.yaml -------------------------------------------------------------------------------- /extract/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extract/extract_cvnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/extract_cvnet.py -------------------------------------------------------------------------------- /extract/extract_descriptors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/extract_descriptors.py -------------------------------------------------------------------------------- /extract/extract_dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/extract_dino.py -------------------------------------------------------------------------------- /extract/image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/image_dataset.py -------------------------------------------------------------------------------- /extract/merge_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/merge_hdf5.py -------------------------------------------------------------------------------- /extract/prepare_topk_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/prepare_topk_global.py -------------------------------------------------------------------------------- /extract/spatial_attention_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/spatial_attention_2d.py -------------------------------------------------------------------------------- /extract/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/extract/transforms.py -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/hubconf.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/ames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/models/ames.py -------------------------------------------------------------------------------- /src/models/binarization_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/models/binarization_layer.py -------------------------------------------------------------------------------- /src/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/models/resnet.py -------------------------------------------------------------------------------- /src/models/transformer_dec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/models/transformer_dec.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/dataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/dataset_loader.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/revisited.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/revisited.py -------------------------------------------------------------------------------- /src/utils/tensor_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/tensor_dataset.py -------------------------------------------------------------------------------- /src/utils/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/training.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pavelsuma/ames/HEAD/src/utils/utils.py --------------------------------------------------------------------------------