https://github.com/kinwaicheuk/pytorch-triplet-loss

Triplet-net and triplet loss implementation on pytorch
https://github.com/kinwaicheuk/pytorch-triplet-loss

Last synced: 3 days ago
JSON representation

Triplet-net and triplet loss implementation on pytorch

• Host: GitHub
• URL: https://github.com/kinwaicheuk/pytorch-triplet-loss
• Owner: KinWaiCheuk
• Created: 2019-03-25T14:04:23.000Z (about 6 years ago)
• Default Branch: master
• Last Pushed: 2022-01-11T21:37:37.000Z (over 3 years ago)
• Last Synced: 2025-03-30T18:51:11.405Z (about 1 month ago)
• Language: Jupyter Notebook
• Size: 14.9 MB
• Stars: 42
• Watchers: 1
• Forks: 9
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# PyTorch Triplet Loss and Online Mining

Implementation of triplet loss, and online mining on Pytorch

This code is a PyTorch implementation of Olivier Moindrot's blog post https://omoindrot.github.io/triplet-loss and his github repository https://github.com/omoindrot/tensorflow-triplet-loss, who implement the triplet loss and online mining on Tensorflow.

My repository aims at implementing triplet loss with minimal amount of codes.

See my Keras version here https://github.com/KinWaiCheuk/Triplet-net-keras (offline mining version)

![alt text](https://github.com/KinWaiCheuk/Triplet-net-pytorch/blob/master/10class.gif?raw=true)

# Dependencies

pytorch >=1.3

sklearn >=0.19.1

matplotlib >=2.2.2

seaborn >=0.9

# Instruction

All the required codes are contained inside the jupyter-notebook