KevinMusgrave/pytorch-metric-learning
Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.
GitHub repository with 6,344 stars and 663 forks.
Language: Python
Topics: computer-vision, contrastive-learning, deep-learning, deep-metric-learning, embeddings, image-retrieval, machine-learning, metric-learning, pytorch, self-supervised-learning