pytorch-metric-learning
similarity
| pytorch-metric-learning | similarity | |
|---|---|---|
| 3 | 7 | |
| 5,770 | 996 | |
| - | 0.2% | |
| 7.9 | 6.5 | |
| about 1 month ago | about 1 month ago | |
| Python | Python | |
| MIT License | Apache License 2.0 |
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.
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
similarity
-
New free tool that uses fine-tuned BERT model to surface answers from research papers
Tensorflow Ranking and Tensorflow similarity (maybe relevant/irrelevant contrastive learning?) look like they could be useful.
-
Non-Machine Learning Image Matching with a Vector DB
There is the metric learning problem to learn a hash for similarity https://github.com/tensorflow/similarity
That said, I don't see many good models available for download on tfhub or huggingface optimized for it, but you can always programmatically modify your images (if you truly mean identical to humans) - change white balance, crop, rotate, select adjacent frames from videos, etc. and optimize a network that is small enough for you to be satisfied and see if that works, as a possible alternative.
-
Face Detection for 520 People
Metric learning has great implementations inside Tensorflow Similarity library: https://github.com/tensorflow/similarity Although the documentation is quite bad, but the jupyter notebooks are great.
-
[P] TensorFlow Similarity 0.16 is out
Just a quick note that TensorFlow Similarity 0.16 is out -- this release beside adding the XMB loss is mostly focus on refactoring and optimizing the core components to ensure everything works smoothly and accurately. Details are in the changelog as usual and a simple pip install -U tensorflow_similarity should just work.
- Self-supervised learning added to TensorFlow Similarity
-
[P] TensorFlow Similarity now self-supervised training
Very happy to announce that as part of the 0.15 release, TensorFlow Similarity now support self-supervised learning using STOA algorithms. To help you get started we included in the release a detailed getting started notebook that you can run in Colab. This notebook shows you how to use SimSiam self-supervised pre-training to almost double the accuracy compared to a model trained from scratch on CIFAR 10.
-
TensorFlow Introduces ‘TensorFlow Similarity’, An Easy And Fast Python Package To Train Similarity Models Using TensorFlow
Github: https://github.com/tensorflow/similarity
What are some alternatives?
dino - PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
pgANN - Fast Approximate Nearest Neighbor (ANN) searches with a PostgreSQL database.
lightly - A python library for self-supervised learning on images.
quaterion - Blazing fast framework for fine-tuning similarity learning models
EasyOCR - Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
ContraD - Code for the paper "Training GANs with Stronger Augmentations via Contrastive Discriminator" (ICLR 2021)
byol-pytorch - Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
Real-Time-Voice-Cloning - Clone a voice in 5 seconds to generate arbitrary speech in real-time
autogluon - Fast and Accurate ML in 3 Lines of Code
sparse_dot_topn - Python package to accelerate the sparse matrix multiplication and top-n similarity selection
simsiam-cifar10 - Code to train the SimSiam model on cifar10 using PyTorch
Keras - Deep Learning for humans