pytorch-metric-learning

Repository facts

Stars

6,342

Forks

660

Open issues and pull requests

85

Primary language

Python

License

MIT License (MIT)

Owner

KevinMusgrave · Personal account

Created

Oct 23, 2019

Last push

Aug 17, 2025

Topics

• computer-vision
• contrastive-learning
• deep-learning
• deep-metric-learning
• embeddings
• image-retrieval
• machine-learning
• metric-learning
• pytorch
• self-supervised-learning

Rank #1,277 of 9,507 in AI on GitHub, by stars. Data from the GitHub API, fetched Oct 18, 2026; star counts change daily. Stars measure popularity, not quality or safety, and a listing is not an endorsement. How this list is made

Related projects

Projects that share this one’s GitHub topics.

• 2,076

  Rank 2,076: lightly-ai/lightly

  A python library for self-supervised learning on images.

  3,821 starsPythonMITLast push Oct 15, 2026

• 4,134

  Rank 4,134: lightly-ai/lightly-train

  All-in-one training for vision models (YOLO, ViTs, RT-DETR, DINOv3): pretraining, fine-tuning, distillation.

  1,715 starsPythonAGPL-3.0Last push Oct 12, 2026

• 8,760

  Rank 8,760: Confusezius/Deep-Metric-Learning-Baselines

  PyTorch Implementation for Deep Metric Learning Pipelines

  576 starsPythonApache-2.0Last push Jun 17, 2020

• 1,635

  Rank 1,635: KaiyangZhou/deep-person-reid

  Torchreid: Deep learning person re-identification in PyTorch.

  4,926 starsPythonMITLast push Jan 9, 2026

• 9,166

  Rank 9,166: DagnyT/hardnet

  Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"

  534 starsPythonMITLast push Apr 14, 2025

• 2,277

  Rank 2,277: towhee-io/towhee

  Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.

  3,450 starsPythonApache-2.0Last push Oct 18, 2024