pytorch-metric-learning
Metric Learning Library
A PyTorch library for implementing deep metric learning algorithms in computer vision applications.
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
6k stars
62 watching
657 forks
Language: Python
last commit: about 2 months ago computer-visioncontrastive-learningdeep-learningdeep-metric-learningembeddingsimage-retrievalmachine-learningmetric-learningpytorchself-supervised-learning
Related projects:
| Repository | Description | Stars |
|---|---|---|
 jfzhang95/pytorch-deeplab-xception | A PyTorch implementation of the DeepLab-V3-Plus model with support for multiple backbones and datasets | 2,919 |
 shangtongzhang/deeprl | A modularized implementation of various deep reinforcement learning algorithms in PyTorch | 3,209 |
 oml-team/open-metric-learning | A PyTorch-based framework for training and validating models that produce high-quality embeddings for computer vision and other tasks. | 897 |
 ahkarami/deep-learning-in-production | A collection of notes and references on deploying deep learning models in production environments | 4,313 |
 p-christ/deep-reinforcement-learning-algorithms-with-pytorch | PyTorch implementations of popular deep reinforcement learning algorithms and environments. | 5,669 |
 mop/bier | This project implements a deep metric learning framework using an adversarial auxiliary loss to improve robustness. | 39 |
 nvidia/apex | Tools for streamlined mixed precision and distributed training in PyTorch | 8,460 |
 bytedance/byteps | A high-performance distributed deep learning framework supporting multiple frameworks and networks | 3,635 |
 qubvel-org/segmentation_models.pytorch | A comprehensive library for training and applying deep learning models for image segmentation | 9,829 |
 ddbourgin/numpy-ml | A collection of machine learning algorithms implemented in NumPy for rapid experimentation and prototyping. | 15,789 |
 mosaicml/composer | A deep learning training framework that abstracts away complexities of distributed training and parallelization. | 5,190 |
 dsgiitr/d2l-pytorch | An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch | 4,256 |
 sktime/pytorch-forecasting | A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures | 4,038 |
 dsksd/deepnlp-models-pytorch | Pytorch implementations of various Deep NLP models from Stanford's cs-224n course | 2,957 |
 luolc/adabound | An optimizer that combines the benefits of Adam and SGD algorithms | 2,908 |