pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
AI summary
Metric Learning Library
A PyTorch library for implementing deep metric learning algorithms in computer vision applications.
- stars
- 6K
- forks
- 657
- watching
- 62
Similar projects
Found by comparing what the projects do, not just their names.
DeepLab Plus Model
A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets
Reinforcement Learning Library
A modularized implementation of various deep reinforcement learning algorithms in PyTorch
Metric learning framework
A PyTorch-based framework for training and validating models that produce high-quality embeddings for computer vision and other tasks.
Model deployment notes
A collection of notes and references on deploying deep learning models in production environments
RL frameworks
PyTorch implementations of popular deep reinforcement learning algorithms and environments.
mop/bier39
Metric learner
This project implements a deep metric learning framework using an adversarial auxiliary loss to improve robustness.
nvidia/apex8.5K
Training toolkit
Tools for streamlined mixed precision and distributed training in PyTorch
bytedance/byteps3.6K
Distributed DL framework
A high-performance distributed deep learning framework supporting multiple frameworks and networks
Segmentation Model Library
A comprehensive library for training and applying deep learning models for image segmentation
ddbourgin/numpy-ml15.8K
ML toolkit
A collection of machine learning algorithms implemented in NumPy for rapid experimentation and prototyping.
Deep learning trainer
A deep learning training framework that abstracts away complexities of distributed training and parallelization.
Deep learning book adaptation
An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch
Time series forecaster
A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures
DeepNLP models
Pytorch implementations of various Deep NLP models from Stanford's cs-224n course
luolc/adabound2.9K
Optimizer
An optimizer that combines the benefits of Adam and SGD algorithms
