← all repositories

adambielski/siamese-triplet

PyTorch library implementing siamese and triplet networks for learning compact embedding vectors from images.

3.2k stars Python ML Frameworks
siamese-triplet
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

This repository provides implementations of siamese and triplet neural network architectures along with custom loss functions for contrastive learning. It includes online pair/triplet mining strategies that automatically select hard positive and negative examples within mini-batches to improve embedding quality. The code provides dataset wrappers for MNIST, batch samplers for balanced training, and a unified fit function for training these embedding networks.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.