MetricLearning with MNIST w/pytorch

try Metric Learning with MNIST!

This is a good read to understand metric learning.

What is this repo?

We try various metric learning with MNIST dataset and visualize results with T-SNE.

Please see the each notebooks for the codes.

Please see: arcface.ipynb

Please see: cosface.ipynb

See: sphereface.ipynb

Please see: circleloss.ipynb

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
.ipynb_checkpoints		.ipynb_checkpoints
__pycache__		__pycache__
data/MNIST		data/MNIST
log		log
README.md		README.md
arcface.ipynb		arcface.ipynb
arcface_bengalli.ipynb		arcface_bengalli.ipynb
autoaugument.py		autoaugument.py
circleloss.ipynb		circleloss.ipynb
cosface.ipynb		cosface.ipynb
metrics.py		metrics.py
saved_weights.pth		saved_weights.pth
sphereface.ipynb		sphereface.ipynb
train_v2.csv		train_v2.csv
tsne.png		tsne.png
tsne_abnormals.png		tsne_abnormals.png
tsne_classification.png		tsne_classification.png