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.
- Install imagemagick
- Go to root directory of the images
- Run
find -name "*pgm" | xargs -I {} convert {} {}.png
This project is updated to be compatible with pytorch 0.4.0
You can find other project requirements in requirements.txt , which you can install using pip install -r requirements.txt