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

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.