├── .gitignore ├── README.md ├── assets ├── motivation.png └── performance.png ├── dataset ├── SOP.py ├── __init__.py ├── base.py ├── cars.py ├── cub.py ├── sampler.py └── utils.py ├── evaluate.py ├── losses.py ├── net └── resnet.py ├── pytorch_metric_learning ├── __init__.py ├── distances │ ├── __init__.py │ ├── base_distance.py │ ├── cosine_similarity.py │ ├── dot_product_similarity.py │ ├── lp_distance.py │ └── snr_distance.py ├── losses │ ├── __init__.py │ ├── angular_loss.py │ ├── arcface_loss.py │ ├── base_metric_loss_function.py │ ├── circle_loss.py │ ├── contrastive_loss.py │ ├── cosface_loss.py │ ├── cross_batch_memory.py │ ├── fast_ap_loss.py │ ├── generic_pair_loss.py │ ├── intra_pair_variance_loss.py │ ├── large_margin_softmax_loss.py │ ├── lifted_structure_loss.py │ ├── margin_loss.py │ ├── mixins.py │ ├── multi_similarity_loss.py │ ├── n_pairs_loss.py │ ├── nca_loss.py │ ├── normalized_softmax_loss.py │ ├── ntxent_loss.py │ ├── proxy_anchor_loss.py │ ├── proxy_losses.py │ ├── signal_to_noise_ratio_losses.py │ ├── soft_triple_loss.py │ ├── sphereface_loss.py │ ├── supcon_loss.py │ ├── triplet_margin_loss.py │ └── tuplet_margin_loss.py ├── miners │ ├── __init__.py │ ├── angular_miner.py │ ├── base_miner.py │ ├── batch_easy_hard_miner.py │ ├── batch_hard_miner.py │ ├── distance_weighted_miner.py │ ├── embeddings_already_packaged_as_triplets.py │ ├── hdc_miner.py │ ├── maximum_loss_miner.py │ ├── multi_similarity_miner.py │ ├── pair_margin_miner.py │ ├── triplet_margin_miner.py │ └── uniform_histogram_miner.py ├── reducers │ ├── __init__.py │ ├── avg_non_zero_reducer.py │ ├── base_reducer.py │ ├── class_weighted_reducer.py │ ├── divisor_reducer.py │ ├── do_nothing_reducer.py │ ├── mean_reducer.py │ ├── multiple_reducers.py │ ├── per_anchor_reducer.py │ └── threshold_reducer.py ├── regularizers │ ├── __init__.py │ ├── base_regularizer.py │ ├── center_invariant_regularizer.py │ ├── lp_regularizer.py │ ├── regular_face_regularizer.py │ ├── sparse_centers_regularizer.py │ └── zero_mean_regularizer.py ├── samplers │ ├── __init__.py │ ├── fixed_set_of_triplets.py │ ├── hierarchical_sampler.py │ ├── m_per_class_sampler.py │ └── tuples_to_weights_sampler.py ├── testers │ ├── __init__.py │ ├── base_tester.py │ ├── global_embedding_space.py │ ├── global_twostream_embedding_space.py │ └── with_same_parent_label.py ├── trainers │ ├── __init__.py │ ├── base_trainer.py │ ├── cascaded_embeddings.py │ ├── deep_adversarial_metric_learning.py │ ├── metric_loss_only.py │ ├── train_with_classifier.py │ ├── twostream_metric_loss.py │ └── unsupervised_embeddings_using_augmentations.py └── utils │ ├── __init__.py │ ├── accuracy_calculator.py │ ├── common_functions.py │ ├── distributed.py │ ├── inference.py │ ├── key_checker.py │ ├── logging_presets.py │ ├── loss_and_miner_utils.py │ ├── loss_tracker.py │ ├── module_with_records.py │ ├── module_with_records_and_reducer.py │ └── stat_utils.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/README.md -------------------------------------------------------------------------------- /assets/motivation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/assets/motivation.png -------------------------------------------------------------------------------- /assets/performance.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/assets/performance.png -------------------------------------------------------------------------------- /dataset/SOP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/SOP.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/base.py -------------------------------------------------------------------------------- /dataset/cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/cars.py -------------------------------------------------------------------------------- /dataset/cub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/cub.py -------------------------------------------------------------------------------- /dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/sampler.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/evaluate.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/losses.py -------------------------------------------------------------------------------- /net/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/net/resnet.py -------------------------------------------------------------------------------- /pytorch_metric_learning/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.9.99" 2 | -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/base_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/base_distance.py -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/cosine_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/cosine_similarity.py -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/dot_product_similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/dot_product_similarity.py -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/lp_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/lp_distance.py -------------------------------------------------------------------------------- /pytorch_metric_learning/distances/snr_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/distances/snr_distance.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/angular_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/angular_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/arcface_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/arcface_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/base_metric_loss_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/base_metric_loss_function.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/circle_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/circle_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/contrastive_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/contrastive_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/cosface_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/cosface_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/cross_batch_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/cross_batch_memory.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/fast_ap_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/fast_ap_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/generic_pair_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/generic_pair_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/intra_pair_variance_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/intra_pair_variance_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/large_margin_softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/large_margin_softmax_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/lifted_structure_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/lifted_structure_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/margin_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/margin_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/mixins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/mixins.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/multi_similarity_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/multi_similarity_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/n_pairs_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/n_pairs_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/nca_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/nca_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/normalized_softmax_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/normalized_softmax_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/ntxent_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/ntxent_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/proxy_anchor_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/proxy_anchor_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/proxy_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/proxy_losses.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/signal_to_noise_ratio_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/signal_to_noise_ratio_losses.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/soft_triple_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/soft_triple_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/sphereface_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/sphereface_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/supcon_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/supcon_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/triplet_margin_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/triplet_margin_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/losses/tuplet_margin_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/losses/tuplet_margin_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/angular_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/angular_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/base_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/base_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/batch_easy_hard_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/batch_easy_hard_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/batch_hard_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/batch_hard_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/distance_weighted_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/distance_weighted_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/embeddings_already_packaged_as_triplets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/embeddings_already_packaged_as_triplets.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/hdc_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/hdc_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/maximum_loss_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/maximum_loss_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/multi_similarity_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/multi_similarity_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/pair_margin_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/pair_margin_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/triplet_margin_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/triplet_margin_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/miners/uniform_histogram_miner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/miners/uniform_histogram_miner.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/avg_non_zero_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/avg_non_zero_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/base_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/base_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/class_weighted_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/class_weighted_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/divisor_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/divisor_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/do_nothing_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/do_nothing_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/mean_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/mean_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/multiple_reducers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/multiple_reducers.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/per_anchor_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/per_anchor_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/reducers/threshold_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/reducers/threshold_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/base_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/base_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/center_invariant_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/center_invariant_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/lp_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/lp_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/regular_face_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/regular_face_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/sparse_centers_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/sparse_centers_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/regularizers/zero_mean_regularizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/regularizers/zero_mean_regularizer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/samplers/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/samplers/fixed_set_of_triplets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/samplers/fixed_set_of_triplets.py -------------------------------------------------------------------------------- /pytorch_metric_learning/samplers/hierarchical_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/samplers/hierarchical_sampler.py -------------------------------------------------------------------------------- /pytorch_metric_learning/samplers/m_per_class_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/samplers/m_per_class_sampler.py -------------------------------------------------------------------------------- /pytorch_metric_learning/samplers/tuples_to_weights_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/samplers/tuples_to_weights_sampler.py -------------------------------------------------------------------------------- /pytorch_metric_learning/testers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/testers/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/testers/base_tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/testers/base_tester.py -------------------------------------------------------------------------------- /pytorch_metric_learning/testers/global_embedding_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/testers/global_embedding_space.py -------------------------------------------------------------------------------- /pytorch_metric_learning/testers/global_twostream_embedding_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/testers/global_twostream_embedding_space.py -------------------------------------------------------------------------------- /pytorch_metric_learning/testers/with_same_parent_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/testers/with_same_parent_label.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/__init__.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/base_trainer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/cascaded_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/cascaded_embeddings.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/deep_adversarial_metric_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/deep_adversarial_metric_learning.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/metric_loss_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/metric_loss_only.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/train_with_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/train_with_classifier.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/twostream_metric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/twostream_metric_loss.py -------------------------------------------------------------------------------- /pytorch_metric_learning/trainers/unsupervised_embeddings_using_augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/trainers/unsupervised_embeddings_using_augmentations.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/accuracy_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/accuracy_calculator.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/common_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/common_functions.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/distributed.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/inference.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/key_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/key_checker.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/logging_presets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/logging_presets.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/loss_and_miner_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/loss_and_miner_utils.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/loss_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/loss_tracker.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/module_with_records.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/module_with_records.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/module_with_records_and_reducer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/module_with_records_and_reducer.py -------------------------------------------------------------------------------- /pytorch_metric_learning/utils/stat_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/pytorch_metric_learning/utils/stat_utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzzheng/IDML/HEAD/utils.py --------------------------------------------------------------------------------