#566 · Primary category: Computer Vision
Deep-Metric-Learning-Baselines
PyTorch Implementation for Deep Metric Learning Pipelines
Project last updated:06/17/20
GitHub Stars
576
Forks
91
Contributors
2
License
Apache-2.0
Why we included this project
Training image embeddings for retrieval or verification usually comes down to picking a loss and a sampling strategy and seeing how they interact. This repository collects the classic deep metric learning losses, triplet, margin, ProxyNCA, and N-pair, along with the sampling methods that pair with them, and runs them on standard fine-grained benchmarks like CUB-200, CARS196, and Stanford Online Products. The code is kept deliberately small, a single training script, a losses module, and dataloaders for the common datasets, so you can trace how a method is actually implemented instead of hunting through a large framework. That makes it useful for researchers who want a clean baseline to compare against and for students trying to understand how metric learning is wired up in PyTorch. One honest caveat: this is a research codebase rather than a maintained production library, and the authors point to a newer repository with the full evaluation suite.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
opencv
Open Source Computer Vision Library
RuView
π RuView turns commodity WiFi signals into real-time spatial intelligence, vital sign monitoring, and presence detection — all without a single pixel of video.
PaddleOCR
Turn any PDF or image document into structured data for your AI. A powerful, lightweight OCR toolkit that bridges the gap between images/PDFs and LLMs. Supports 100+ languages.
MinerU
Transforms complex documents like PDFs and Office docs into LLM-ready markdown/JSON for your Agentic workflows.
tesseract
Tesseract Open Source OCR Engine (main repository)