├── .gitignore ├── LICENSE ├── README.md ├── browse_results.py ├── collect_env_info.py ├── dataset ├── __init__.py ├── bytebox │ ├── __init__.py │ ├── base.py │ ├── cars.py │ ├── cub.py │ ├── inshop.py │ ├── sop.py │ ├── utils.py │ └── vid.py ├── loader.py ├── npairs.py └── sampler.py ├── eval_model.py ├── experiment ├── __init__.py ├── base.py └── margin_loss_resnet50.py ├── metriclearning ├── .gitignore ├── __init__.py ├── evaluation │ ├── __init__.py │ ├── mean_avg_precision.py │ ├── normalized_mutual_information.py │ └── recall.py ├── faissext.py ├── loss │ ├── __init__.py │ ├── margin_loss.py │ ├── npairs_loss.py │ └── triplet_loss.py ├── model │ ├── __init__.py │ ├── bn_inception.py │ ├── inweave │ │ ├── __init__.py │ │ ├── init.py │ │ └── models.py │ ├── resnet.py │ └── torchvision_models.py ├── sampler │ ├── __init__.py │ ├── pairs.py │ └── triplets.py ├── similarity.py └── utils.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/README.md -------------------------------------------------------------------------------- /browse_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/browse_results.py -------------------------------------------------------------------------------- /collect_env_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/collect_env_info.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/bytebox/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/__init__.py -------------------------------------------------------------------------------- /dataset/bytebox/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/base.py -------------------------------------------------------------------------------- /dataset/bytebox/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/cars.py -------------------------------------------------------------------------------- /dataset/bytebox/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/cub.py -------------------------------------------------------------------------------- /dataset/bytebox/inshop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/inshop.py -------------------------------------------------------------------------------- /dataset/bytebox/sop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/sop.py -------------------------------------------------------------------------------- /dataset/bytebox/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/utils.py -------------------------------------------------------------------------------- /dataset/bytebox/vid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/bytebox/vid.py -------------------------------------------------------------------------------- /dataset/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/loader.py -------------------------------------------------------------------------------- /dataset/npairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/npairs.py -------------------------------------------------------------------------------- /dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/dataset/sampler.py -------------------------------------------------------------------------------- /eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/eval_model.py -------------------------------------------------------------------------------- /experiment/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/experiment/__init__.py -------------------------------------------------------------------------------- /experiment/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/experiment/base.py -------------------------------------------------------------------------------- /experiment/margin_loss_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/experiment/margin_loss_resnet50.py -------------------------------------------------------------------------------- /metriclearning/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/.gitignore -------------------------------------------------------------------------------- /metriclearning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/__init__.py -------------------------------------------------------------------------------- /metriclearning/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/evaluation/__init__.py -------------------------------------------------------------------------------- /metriclearning/evaluation/mean_avg_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/evaluation/mean_avg_precision.py -------------------------------------------------------------------------------- /metriclearning/evaluation/normalized_mutual_information.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/evaluation/normalized_mutual_information.py -------------------------------------------------------------------------------- /metriclearning/evaluation/recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/evaluation/recall.py -------------------------------------------------------------------------------- /metriclearning/faissext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/faissext.py -------------------------------------------------------------------------------- /metriclearning/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/loss/__init__.py -------------------------------------------------------------------------------- /metriclearning/loss/margin_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/loss/margin_loss.py -------------------------------------------------------------------------------- /metriclearning/loss/npairs_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/loss/npairs_loss.py -------------------------------------------------------------------------------- /metriclearning/loss/triplet_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/loss/triplet_loss.py -------------------------------------------------------------------------------- /metriclearning/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/__init__.py -------------------------------------------------------------------------------- /metriclearning/model/bn_inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/bn_inception.py -------------------------------------------------------------------------------- /metriclearning/model/inweave/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/inweave/__init__.py -------------------------------------------------------------------------------- /metriclearning/model/inweave/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/inweave/init.py -------------------------------------------------------------------------------- /metriclearning/model/inweave/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/inweave/models.py -------------------------------------------------------------------------------- /metriclearning/model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/resnet.py -------------------------------------------------------------------------------- /metriclearning/model/torchvision_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/model/torchvision_models.py -------------------------------------------------------------------------------- /metriclearning/sampler/__init__.py: -------------------------------------------------------------------------------- 1 | from .triplets import * 2 | -------------------------------------------------------------------------------- /metriclearning/sampler/pairs.py: -------------------------------------------------------------------------------- 1 | # for backward compatibility 2 | from .triplets import * 3 | -------------------------------------------------------------------------------- /metriclearning/sampler/triplets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/sampler/triplets.py -------------------------------------------------------------------------------- /metriclearning/similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/similarity.py -------------------------------------------------------------------------------- /metriclearning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/metriclearning/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CompVis/metric-learning-divide-and-conquer-improved/HEAD/utils.py --------------------------------------------------------------------------------