KevinMusgrave/pytorch-metric-learning
A PyTorch library for deep metric learning that provides losses, miners, and trainers for learning similarity-preserving embeddings.

Velocity · 7d
+2.6
★ / day
Trend
→steady
star history
This library provides modular implementations of metric learning algorithms including contrastive learning, triplet loss variants, and ArcFace-style losses. It offers components for data mining (hard negative mining), logging, and evaluation specifically for embedding-based retrieval tasks. The library integrates with standard CV datasets and includes Colab examples for common image retrieval benchmarks.