P
pytorch-metric-learning
KevinMusgrave/pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
★6.3kstars
Python
MIT
Updated: 1w ago
📋 Project at a Glance
Tap to expand
What's this?A , built with Python open-source project in the Data & Infrastructure category, core strengths: computer-vision/contrastive-learning
Who made it?Maintained by KevinMusgrave team, 6.3K⭐ on GitHub, #295 out of 3133 in Data & Infrastructure
Why does it exist?As the Data & Infrastructure landscape evolved, the KevinMusgrave team identified the need for better computer-vision solutions. pytorch-metric-learning was created to simplify contrastive-learning workflows.
What can it do?Key use cases: deep-learning, deep-metric-learning, embeddings
How to install with AI?Use an AI coding assistant (Claude Code, Cursor, Copilot) to automatically set up pip dependencies and virtual env. Follow the README — the AI handles the rest.
🔗 github.com/KevinMusgrave/pytorch-metric-learning | 官网 https://kevinmusgrave.github.io/pytorch-metric-learning/
🔗 github.com/KevinMusgrave/pytorch-metric-learning | 官网 https://kevinmusgrave.github.io/pytorch-metric-learning/
Topics
computer-visioncontrastive-learningdeep-learningdeep-metric-learningembeddingsimage-retrievalmachine-learningmetric-learningpytorchself-supervised-learning