DIML
pytorch-metric-learning
DIML | pytorch-metric-learning | |
---|---|---|
1 | 3 | |
83 | 5,785 | |
- | - | |
5.2 | 7.9 | |
over 2 years ago | about 1 month ago | |
Python | Python | |
- | MIT License |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
DIML
pytorch-metric-learning
-
Similarity Learning lacks a framework. So we built one
Not a full featured framework, but pytorch-metric-learning has data loaders, lossess, etc. to facilitate similarity learning: https://github.com/KevinMusgrave/pytorch-metric-learning
Disclaimer: I've made some contributions to it.
-
[R][D] VAE Embedding Space - Can we force it to learn a metric?
You can use the triplet loss together with the Gaussian prior. It will be zero centered though and the clusters are not as separated when you use the triplet loss only.There are many alternative to the triplet loss, in case it needs to be a metric: https://github.com/KevinMusgrave/pytorch-metric-learning
-
[D] Similar Image Retrieval
This repo provides the tools and examples needed to build such a model: https://github.com/KevinMusgrave/pytorch-metric-learning
What are some alternatives?
d2l-en - Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
dino - PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
XVFI - [ICCV 2021, Oral 3%] Official repository of XVFI
lightly - A python library for self-supervised learning on images.
surface_normal_uncertainty - (ICCV 2021 - oral) Estimating and Exploiting the Aleatoric Uncertainty in Surface Normal Estimation
EasyOCR - Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
pytorch-grad-cam - Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
byol-pytorch - Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
autogluon - Fast and Accurate ML in 3 Lines of Code
simsiam-cifar10 - Code to train the SimSiam model on cifar10 using PyTorch
similarity - TensorFlow Similarity is a python package focused on making similarity learning quick and easy.
Transformer-SSL - This is an official implementation for "Self-Supervised Learning with Swin Transformers".