Awesome Open Source
Awesome Open Source

Facial Similarity with Siamese Networks in Pytorch

You can read the accompanying article at https://hackernoon.com/one-shot-learning-with-siamese-networks-in-pytorch-8ddaab10340e

The goal is to teach a siamese network to be able to distinguish pairs of images. This project uses pytorch.

Any dataset can be used. Each class must be in its own folder. This is the same structure that PyTorch's own image folder dataset uses.

Update: Looking for contributor (July 2020)

If you would like to be a part of this projec, please head to https://github.com/harveyslash/Facial-Similarity-with-Siamese-Networks-in-Pytorch/issues/35

Converting pgm files (if you decide to use the AT&T dataset) to png

  1. Install imagemagick
  2. Go to root directory of the images
  3. Run find -name "*pgm" | xargs -I {} convert {} {}.png

Installing the right version of PyTorch

This project is updated to be compatible with pytorch 0.4.0

This project requires python3.6

Alternatives To Facial Similarity With Siamese Networks In Pytorch
Select To Compare


Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Jupyter Notebook (171,617
Deep Learning (38,604
Network (37,504
Dataset (33,181
Pytorch (22,212
Neural Network (15,945
Similarity (3,744
Face Recognition (2,531
Pytorch Tutorial (320
Siamese Network (255