Skip to content

neka-nat/pytorch-softtriple

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SoftTriple

This is an unofficial implementation of "SoftTriple Loss: Deep Metric Learning Without Triplet Sampling" (ICCV 2019) in Pytorch.

Installation

cd pytorch-hdml
pip install pipenv
pipenv install

Download dataset

cd data
python cars196_downloader.py
python cars196_converter.py

Train CARS196 dataset

Execute a training script. When executed, the tensorboard log is saved.

pipenv shell
python train_softtriple.py

Result

CARS196 result on training(99 classes, 30000 iterations)

Loss

loss

t-SNE

train_tsne

Releases

No releases published

Packages

No packages published

Languages