pytorch-metric-learning
Metric Learning Library
A PyTorch-based library for easy implementation of deep metric learning in applications.
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
6k stars
62 watching
658 forks
Language: Python
last commit: 21 days 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,910 |
 shangtongzhang/deeprl | A modularized implementation of various deep reinforcement learning algorithms in PyTorch | 3,191 |
 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. | 882 |
 ahkarami/deep-learning-in-production | A collection of notes and references on deploying deep learning models in production environments | 4,308 |
 p-christ/deep-reinforcement-learning-algorithms-with-pytorch | PyTorch implementations of popular deep reinforcement learning algorithms and environments. | 5,640 |
 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,417 |
 bytedance/byteps | A high-performance distributed deep learning framework supporting multiple frameworks and networks | 3,630 |
 qubvel-org/segmentation_models.pytorch | A PyTorch library for building and training neural networks for image segmentation tasks. | 9,696 |
 ddbourgin/numpy-ml | A collection of machine learning algorithms implemented in NumPy for rapid experimentation and prototyping. | 15,466 |
 mosaicml/composer | A PyTorch-based deep learning training library designed to simplify large-scale model training on clusters. | 5,165 |
 dsgiitr/d2l-pytorch | An open-source implementation of the popular book 'Dive Into Deep Learning' in PyTorch | 4,244 |
 sktime/pytorch-forecasting | A PyTorch-based package for state-of-the-art time series forecasting with deep learning architectures | 4,001 |
 dsksd/deepnlp-models-pytorch | Pytorch implementations of various Deep NLP models from Stanford's cs-224n course | 2,954 |
 luolc/adabound | An optimizer that combines the benefits of Adam and SGD algorithms | 2,907 |