metric-learning

There are 23 repositories under metric-learning topic.

pytorch-metric-learning
KevinMusgrave / pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
computer-vision contrastive-learning deep-learning deep-metric-learning embeddings image-retrieval machine-learning metric-learning pytorch self-supervised-learning
Language:Python 6249
KaiyangZhou / deep-person-reid
Torchreid: Deep learning person re-identification in PyTorch.
computer-vision cross-domain deep-learning deep-neural-networks image-retrieval machine-learning metric-learning person-reid person-reidentification pytorch re-ranking
Language:Python 4657
Person_reID_baseline_pytorch
layumi / Person_reID_baseline_pytorch
:bouncing_ball_person: Pytorch ReID: A tiny, friendly, strong pytorch implement of person re-id / vehicle re-id baseline. Tutorial 👉https://github.com/layumi/Person_reID_baseline_pytorch/tree/master/tutorial
awesome-reid baseline circle-loss cuhk-np image-retrieval image-search market-1501 metric-learning msmt17 object-reid open-reid person-reid person-reidentification pytorch random-erasing re-ranking tutorial vehicle-reid
Language:Python 4357
catalyst-team / catalyst
Accelerated deep learning R&D
computer-vision deep-learning distributed-computing image-classification image-processing image-segmentation information-retrieval infrastructure machine-learning metric-learning natural-language-processing object-detection python pytorch recommender-system reinforcement-learning reproducibility research text-classification text-segmentation
Language:Python 3362
finetuner
jina-ai / finetuner
:dart: Task-oriented embedding tuning for BERT, CLIP, etc.
fine-tuning pretrained-models few-shot-learning negative-sampling metric-learning siamese-network triplet-loss transfer-learning jina neural-search finetuning similarity-learning bert openai-clip
Language:Python 1506
scikit-learn-contrib / metric-learn
Metric learning algorithms in Python
python metric-learning scikit-learn machine-learning
Language:Python 1426
Cysu / open-reid
Open source person re-identification library in python
person-reidentification deep-learning metric-learning pytorch python
Language:Python 1367
jakesnell / prototypical-networks
Code for the NeurIPS 2017 Paper "Prototypical Networks for Few-shot Learning"
deep-learning few-shot metric-learning nips-2017 omniglot pytorch
Language:Python 1203
clovaai / voxceleb_trainer
In defence of metric learning for speaker recognition
metric-learning speaker-recognition speaker-verification voxceleb
Language:Python 1143
tensorflow / similarity
TensorFlow Similarity is a python package focused on making similarity learning quick and easy.
similarity-learning metric-learning similarity-search nearest-neighbor-search nearest-neighbors deep-learning tensorflow contrastive-learning machine-learning unsupervised-learning clustering python simclr simclr2 barlow-twins simsiam cosine-similarity knn
Language:Python 1022
OML-Team / open-metric-learning
Metric learning and retrieval pipelines, models and zoo.
computer-vision data-science deep-learning metric-learning pytorch pytorch-lightning representation-learning hacktoberfest similarity-learning hacktoberfest-2023 hacktoberfest2023
Language:Python 977
quaterion
qdrant / quaterion
Blazing fast framework for fine-tuning similarity learning models
contrastive-learning cosine-similarity deep-learning knn machine-learning metric-learning nearest-neighbor-search python pytorch pytorch-lightning similarity-learning similarity-search
Language:Python 661
earhian / Humpback-Whale-Identification-1st-
https://www.kaggle.com/c/humpback-whale-identification
classification deep-learning kaggle-competition metric-learning
Language:Python 641
Confusezius / Deep-Metric-Learning-Baselines
PyTorch Implementation for Deep Metric Learning Pipelines
cars196 computer-vision cub200 deep-learning deep-metric-learning distance-sampling metric-learning neural-networks pku-vehicle pytorch shop-clothes
Language:Python 574
ryanzhumich / Contrastive-Learning-NLP-Papers
Paper List for Contrastive Learning for Natural Language Processing
contrastive-learning machine-learning metric-learning natural-language-processing representation-learning
563
DagnyT / hardnet
Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"
deep-learning convolutional-neural-networks computer-vision metric-learning image-matching image-retrieval pytorch nips-2017
Language:Python 522
qdrant / awesome-metric-learning
😎 A curated list of awesome practical Metric Learning and its applications
metric-learning awesome-list tutorials survey recommendation-system anomaly-detection semantic-similarity similarity-search
515
cvqluu / Angular-Penalty-Softmax-Losses-Pytorch
Angular penalty loss functions in Pytorch (ArcFace, SphereFace, Additive Margin, CosFace)
metric-learning pytorch loss-functions loss-function embedding face-verification fashion-mnist fmnist-dataset face-recognition speaker-recognition sphereface arcface normface am-softmax
Language:Python 494
Joker316701882 / Additive-Margin-Softmax
This is the implementation of paper <Additive Margin Softmax for Face Verification>
deeplearning facerecognition metric-learning sphereface am-softmax
Language:Jupyter Notebook 494
happynear / AMSoftmax
A simple yet effective loss function for face verification.
deep-learning face-recognition loss-functions metric-learning softmax
Language:MATLAB 489
powerful-benchmarker
KevinMusgrave / powerful-benchmarker
A library for ML benchmarking. It's powerful.
pytorch benchmarking computer-vision machine-learning deep-learning domain-adaptation metric-learning transfer-learning
Language:Jupyter Notebook 437
lenscloth / RKD
Official pytorch Implementation of Relational Knowledge Distillation, CVPR 2019
deep-neural-networks computer-vision metric-learning knowledge-distillation deep-learning
Language:Python 405
JohnGiorgi / DeCLUTR
The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!
contrastive-learning natural-language-processing allennlp pytorch transformers representation-learning sentence-embeddings sentence-similarity semantic-search semantic-text-similarity metric-learning self-supervised-learning
Language:Python 380
tfzhou / ProtoSeg
CVPR2022 (Oral) - Rethinking Semantic Segmentation: A Prototype View
clustering deep-learning fcn metric-learning nearest-neighbours-classifier nonparametric prototype segmentation semantic-segmentation softmax transformer
Language:Python 378
Revisiting_Deep_Metric_Learning_PyTorch
Confusezius / Revisiting_Deep_Metric_Learning_PyTorch
(ICML 2020) This repo contains code for our paper "Revisiting Training Strategies and Generalization Performance in Deep Metric Learning" (https://arxiv.org/abs/2002.08473) to facilitate consistent research in the field of Deep Metric Learning.
cars196 cub200-2011 deep-learning deep-metric-learning icml2020 image-retrieval metric-learning stanford-online-products wandb
Language:Python 345
daizuozhuo / batch-dropblock-network
Official source code of "Batch DropBlock Network for Person Re-identification and Beyond" (ICCV 2019)
image-retrieval metric-learning person-reidentification pytorch
Language:Python 327
leeesangwon / PyTorch-Image-Retrieval
A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).
metric-learning n-pair-loss angular-loss image-retrieval pytorch deep-metric-learning
Language:Python 282
CompVis / metric-learning-divide-and-conquer
Source code for the paper "Divide and Conquer the Embedding Space for Metric Learning", CVPR 2019
few-shot-learning metric-learning pytorch
Language:Python 265
vithursant / MagnetLoss-PyTorch
PyTorch implementation of a deep metric learning technique called "Magnet Loss" from Facebook AI Research (FAIR) in ICLR 2016.
metric-learning cnn pytorch classification docker deep-learning embeddings anaconda
Language:Python 219
ShengcaiLiao / QAConv
[ECCV 2020] QAConv: Interpretable and Generalizable Person Re-Identification with Query-Adaptive Convolution and Temporal Lifting, and [CVPR 2022] GS: Graph Sampling Based Deep Metric Learning
person-reidentification person-re-identification person-reid person-retrieval person-search person-recognition re-identification re-id temporal-models adaptive-convolution interpretability interpretable-deep-learning correspondence image-matching generalization generalizability deep-metric-learning metric-learning reid domain-generalization
Language:Python 214
cambridgeltl / sapbert
[NAACL'21 & ACL'21] SapBERT: Self-alignment pretraining for BERT & XL-BEL: Cross-Lingual Biomedical Entity Linking.
nlp machine-learning language-model lexical-semantics representation-learning naacl2021 acl2021 bionlp bert contrastive-learning metric-learning
Language:Python 209
jeromerony / dml_cross_entropy
Code for the paper "A unifying mutual information view of metric learning: cross-entropy vs. pairwise losses" (ECCV 2020 - Spotlight)
metric-learning deep-learning cross-entropy
Language:Python 169
BenevolentAI / DeeplyTough
DeeplyTough: Learning Structural Comparison of Protein Binding Sites
protein-structure drug-discovery deep-learning metric-learning 3d-models
Language:Python 166
ronekko / deep_metric_learning
Deep metric learning methods implemented in Chainer
deep-learning python chainer metric-learning
Language:Python 158
jyhjinghwang / SegSort
SegSort: Segmentation by Discriminative Sorting of Segments
computer-vision deep-learning metric-learning self-supervised-learning semantic-segmentation
Language:Python 157
grib0ed0v / face_recognition.pytorch
Deep Face Recognition in PyTorch
face-recognition deep-learning computer-vision landmark-detection deep-face-recognition pytorch arcface cosface focal-loss metric-learning am-softmax center-loss vggface2 lfw sphereface sv-softmax mobilefacenet
Language:Python 153

metric-learning

KevinMusgrave / pytorch-metric-learning

KaiyangZhou / deep-person-reid

layumi / Person_reID_baseline_pytorch

catalyst-team / catalyst

jina-ai / finetuner

scikit-learn-contrib / metric-learn

Cysu / open-reid

jakesnell / prototypical-networks

clovaai / voxceleb_trainer

tensorflow / similarity

OML-Team / open-metric-learning

qdrant / quaterion

earhian / Humpback-Whale-Identification-1st-

Confusezius / Deep-Metric-Learning-Baselines

ryanzhumich / Contrastive-Learning-NLP-Papers

DagnyT / hardnet

qdrant / awesome-metric-learning

cvqluu / Angular-Penalty-Softmax-Losses-Pytorch

Joker316701882 / Additive-Margin-Softmax

happynear / AMSoftmax

KevinMusgrave / powerful-benchmarker

lenscloth / RKD

JohnGiorgi / DeCLUTR

tfzhou / ProtoSeg

Confusezius / Revisiting_Deep_Metric_Learning_PyTorch

daizuozhuo / batch-dropblock-network

leeesangwon / PyTorch-Image-Retrieval

CompVis / metric-learning-divide-and-conquer

vithursant / MagnetLoss-PyTorch

ShengcaiLiao / QAConv

cambridgeltl / sapbert

jeromerony / dml_cross_entropy

BenevolentAI / DeeplyTough

ronekko / deep_metric_learning

jyhjinghwang / SegSort

grib0ed0v / face_recognition.pytorch