Skip to content

A Siamese Network implementation in Pytorch, with additional pytorch-lightning support for training

License

Notifications You must be signed in to change notification settings

LawJarp-A/siamese-network-pytorch

Repository files navigation

siamese-network-pytorch

A Siamese Network implementation in Pytorch, with additional pytorch-lightning support for training

Simple Siamese network

  • Contains code on how to implement a Pytorch Dataset class to get images to train a Siamese network
  • Contains code on how to implemnet a Siamese Network model to train, wrapped inpytorch-lightning

Siamese network with triplet mining

  • Triplet minining is a technique used to increse efficieny and converging rate with triplet loss

About

A Siamese Network implementation in Pytorch, with additional pytorch-lightning support for training

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages