← 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.

994 stars Python ML FrameworksComputer Vision
open-metric-learning
Not currently ranked — collecting fresh signals.
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.

Frequently asked

What is OML-Team/open-metric-learning?
A PyTorch framework for training and validating models that produce high-quality embeddings for metric and similarity learning tasks.
Is open-metric-learning open source?
Yes — OML-Team/open-metric-learning is open source, released under the Apache-2.0 license.
What language is open-metric-learning written in?
OML-Team/open-metric-learning is primarily written in Python.
How popular is open-metric-learning?
OML-Team/open-metric-learning has 994 stars on GitHub.
Where can I find open-metric-learning?
OML-Team/open-metric-learning is on GitHub at https://github.com/OML-Team/open-metric-learning.

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