open-metric-learning
Metric learning framework
A PyTorch-based framework for training and validating models that produce high-quality embeddings for computer vision and other tasks.
Metric learning and retrieval pipelines, models and zoo.
897 stars
11 watching
63 forks
Language: Python
last commit: about 1 year ago
Linked from 1 awesome list
computer-visiondata-sciencedeep-learninghacktoberfesthacktoberfest-2023hacktoberfest2023metric-learningpytorchpytorch-lightningrepresentation-learningsimilarity-learning
josephmisiti/awesome-machine-learningRelated projects:
| Repository | Description | Stars |
|---|---|---|
 open-mmlab/mmengine | Provides a flexible and configurable framework for training deep learning models with PyTorch. | 1,196 |
 scikit-learn-contrib/metric-learn | A Python library providing efficient implementations of various supervised and weakly-supervised metric learning algorithms. | 1,402 |
 enochkan/torch-metrics | A collection of common machine learning evaluation metrics implemented in PyTorch | 110 |
 tristandeleu/pytorch-maml-rl | Replication of Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks in PyTorch for reinforcement learning tasks | 830 |
 erotemic/netharn | A PyTorch framework for managing and automating deep learning training loops with features like hyperparameter tracking and single-file deployments. | 40 |
 catalyst-team/catalyst-rl | A PyTorch framework for accelerating reinforcement learning research and development by providing a modular, reusable, and customizable training loop | 46 |
 kengz/slm-lab | A comprehensive framework for deep reinforcement learning using PyTorch. | 1,256 |
 mop/bier | This project implements a deep metric learning framework using an adversarial auxiliary loss to improve robustness. | 39 |
 pyg-team/pytorch-frame | A deep learning framework for handling heterogeneous tabular data with diverse column types | 582 |
 iffix/machin | An open-source reinforcement learning library for PyTorch, providing a simple and clear implementation of various algorithms. | 402 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,236 |
 subho406/omninet | An implementation of a unified architecture for multi-modal multi-task learning using PyTorch. | 515 |
 graal-research/poutyne | A PyTorch framework simplifying neural network training with automated boilerplate code and callback utilities | 572 |
 matthiasplappert/keras-rl | A Python library implementing state-of-the-art deep reinforcement learning algorithms for Keras and OpenAI Gym environments. | 8 |
 pouyamghari/pof-mkl | An implementation of an online federated learning algorithm with multiple kernels for personalized machine learning | 0 |