├── .gitignore ├── LICENSE ├── README.md ├── Sample_Runs └── Sample_Runs.sh ├── architectures ├── __init__.py ├── bninception.py ├── googlenet.py └── resnet50.py ├── batchminer ├── __init__.py ├── distance.py ├── diva_shared_distance_an.py ├── diva_shared_distance_apn.py ├── easypositive.py ├── epdn.py ├── ephn.py ├── epshn.py ├── intra_random.py ├── lifted.py ├── npair.py ├── pads.py ├── parametric.py ├── random.py ├── random_distance.py ├── rho_distance.py ├── semihard.py └── softhard.py ├── criteria ├── __init__.py ├── adversarial_separation.py ├── angular.py ├── arcface.py ├── contrastive.py ├── histogram.py ├── lifted.py ├── margin.py ├── multisimilarity.py ├── npair.py ├── proxynca.py ├── quadruplet.py ├── s2sd.py ├── snr.py ├── softmax.py ├── softtriplet.py └── triplet.py ├── datasampler ├── __init__.py ├── class_random_sampler.py ├── d2_coreset_sampler.py ├── disthist_batchmatch_sampler.py ├── fid_batchmatch_sampler.py ├── greedy_coreset_sampler.py ├── random_sampler.py └── samplers.py ├── datasets ├── __init__.py ├── basic_dataset_scaffold.py ├── cars196.py ├── cub200.py ├── inshop.py └── stanford_online_products.py ├── evaluation ├── __init__.py └── evaluate.py ├── images ├── rel_comp.png ├── setup.png └── sota_comp.png ├── main.py ├── metrics ├── __init__.py ├── a_recall.py ├── c_f1.py ├── c_mAP_1000.py ├── c_mAP_c.py ├── c_mAP_lim.py ├── c_nmi.py ├── c_recall.py ├── compute_stack.py ├── dists.py ├── e_recall.py ├── f1.py ├── mAP.py ├── mAP_1000.py ├── mAP_c.py ├── mAP_lim.py ├── nmi.py └── rho_spectrum.py ├── parameters.py └── utilities ├── __init__.py ├── logger.py ├── misc.py └── visualize_tsne.py /.gitignore: -------------------------------------------------------------------------------- 1 | *__pycache__ 2 | *.pyc 3 | *Training_Results 4 | *wandb 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/README.md -------------------------------------------------------------------------------- /Sample_Runs/Sample_Runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/Sample_Runs/Sample_Runs.sh -------------------------------------------------------------------------------- /architectures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/architectures/__init__.py -------------------------------------------------------------------------------- /architectures/bninception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/architectures/bninception.py -------------------------------------------------------------------------------- /architectures/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/architectures/googlenet.py -------------------------------------------------------------------------------- /architectures/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/architectures/resnet50.py -------------------------------------------------------------------------------- /batchminer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/__init__.py -------------------------------------------------------------------------------- /batchminer/distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/distance.py -------------------------------------------------------------------------------- /batchminer/diva_shared_distance_an.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/diva_shared_distance_an.py -------------------------------------------------------------------------------- /batchminer/diva_shared_distance_apn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/diva_shared_distance_apn.py -------------------------------------------------------------------------------- /batchminer/easypositive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/easypositive.py -------------------------------------------------------------------------------- /batchminer/epdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/epdn.py -------------------------------------------------------------------------------- /batchminer/ephn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/ephn.py -------------------------------------------------------------------------------- /batchminer/epshn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/epshn.py -------------------------------------------------------------------------------- /batchminer/intra_random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/intra_random.py -------------------------------------------------------------------------------- /batchminer/lifted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/lifted.py -------------------------------------------------------------------------------- /batchminer/npair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/npair.py -------------------------------------------------------------------------------- /batchminer/pads.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /batchminer/parametric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/parametric.py -------------------------------------------------------------------------------- /batchminer/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/random.py -------------------------------------------------------------------------------- /batchminer/random_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/random_distance.py -------------------------------------------------------------------------------- /batchminer/rho_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/rho_distance.py -------------------------------------------------------------------------------- /batchminer/semihard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/semihard.py -------------------------------------------------------------------------------- /batchminer/softhard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/batchminer/softhard.py -------------------------------------------------------------------------------- /criteria/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/__init__.py -------------------------------------------------------------------------------- /criteria/adversarial_separation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/adversarial_separation.py -------------------------------------------------------------------------------- /criteria/angular.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/angular.py -------------------------------------------------------------------------------- /criteria/arcface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/arcface.py -------------------------------------------------------------------------------- /criteria/contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/contrastive.py -------------------------------------------------------------------------------- /criteria/histogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/histogram.py -------------------------------------------------------------------------------- /criteria/lifted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/lifted.py -------------------------------------------------------------------------------- /criteria/margin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/margin.py -------------------------------------------------------------------------------- /criteria/multisimilarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/multisimilarity.py -------------------------------------------------------------------------------- /criteria/npair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/npair.py -------------------------------------------------------------------------------- /criteria/proxynca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/proxynca.py -------------------------------------------------------------------------------- /criteria/quadruplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/quadruplet.py -------------------------------------------------------------------------------- /criteria/s2sd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/s2sd.py -------------------------------------------------------------------------------- /criteria/snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/snr.py -------------------------------------------------------------------------------- /criteria/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/softmax.py -------------------------------------------------------------------------------- /criteria/softtriplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/softtriplet.py -------------------------------------------------------------------------------- /criteria/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/criteria/triplet.py -------------------------------------------------------------------------------- /datasampler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/__init__.py -------------------------------------------------------------------------------- /datasampler/class_random_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/class_random_sampler.py -------------------------------------------------------------------------------- /datasampler/d2_coreset_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/d2_coreset_sampler.py -------------------------------------------------------------------------------- /datasampler/disthist_batchmatch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/disthist_batchmatch_sampler.py -------------------------------------------------------------------------------- /datasampler/fid_batchmatch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/fid_batchmatch_sampler.py -------------------------------------------------------------------------------- /datasampler/greedy_coreset_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/greedy_coreset_sampler.py -------------------------------------------------------------------------------- /datasampler/random_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/random_sampler.py -------------------------------------------------------------------------------- /datasampler/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasampler/samplers.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/basic_dataset_scaffold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/basic_dataset_scaffold.py -------------------------------------------------------------------------------- /datasets/cars196.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/cars196.py -------------------------------------------------------------------------------- /datasets/cub200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/cub200.py -------------------------------------------------------------------------------- /datasets/inshop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/inshop.py -------------------------------------------------------------------------------- /datasets/stanford_online_products.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/datasets/stanford_online_products.py -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/evaluation/evaluate.py -------------------------------------------------------------------------------- /images/rel_comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/images/rel_comp.png -------------------------------------------------------------------------------- /images/setup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/images/setup.png -------------------------------------------------------------------------------- /images/sota_comp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/images/sota_comp.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/main.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/a_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/a_recall.py -------------------------------------------------------------------------------- /metrics/c_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_f1.py -------------------------------------------------------------------------------- /metrics/c_mAP_1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_mAP_1000.py -------------------------------------------------------------------------------- /metrics/c_mAP_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_mAP_c.py -------------------------------------------------------------------------------- /metrics/c_mAP_lim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_mAP_lim.py -------------------------------------------------------------------------------- /metrics/c_nmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_nmi.py -------------------------------------------------------------------------------- /metrics/c_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/c_recall.py -------------------------------------------------------------------------------- /metrics/compute_stack.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /metrics/dists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/dists.py -------------------------------------------------------------------------------- /metrics/e_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/e_recall.py -------------------------------------------------------------------------------- /metrics/f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/f1.py -------------------------------------------------------------------------------- /metrics/mAP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/mAP.py -------------------------------------------------------------------------------- /metrics/mAP_1000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/mAP_1000.py -------------------------------------------------------------------------------- /metrics/mAP_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/mAP_c.py -------------------------------------------------------------------------------- /metrics/mAP_lim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/mAP_lim.py -------------------------------------------------------------------------------- /metrics/nmi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/nmi.py -------------------------------------------------------------------------------- /metrics/rho_spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/metrics/rho_spectrum.py -------------------------------------------------------------------------------- /parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/parameters.py -------------------------------------------------------------------------------- /utilities/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utilities/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/utilities/logger.py -------------------------------------------------------------------------------- /utilities/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/utilities/misc.py -------------------------------------------------------------------------------- /utilities/visualize_tsne.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MLforHealth/S2SD/HEAD/utilities/visualize_tsne.py --------------------------------------------------------------------------------