← all repositories

KevinMusgrave/pytorch-metric-learning

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

6.3k stars Python ML FrameworksComputer Vision
pytorch-metric-learning
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.