← all repositories

OML-Team/open-metric-learning

A PyTorch framework for training and validating models that produce high-quality embeddings for metric and similarity learning tasks.

990 stars Python ML FrameworksComputer Vision
open-metric-learning
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

Open Metric Learning provides pipelines and utilities for training deep learning models that generate embeddings suitable for retrieval, similarity search, and clustering. It includes standard losses like triplet and contrastive losses, pre-built model architectures, and dataset loaders optimized for metric learning workflows. The framework is used for computer vision tasks including image retrieval, re-identification, and verification.

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