Skip to content
This repository has been archived by the owner on Aug 30, 2018. It is now read-only.

hmishfaq/deep-metric-learning

Repository files navigation

How to run:

  1. Install visdom (https://github.com/facebookresearch/visdom)
  2. Run train.py

A PyTorch Implementation for Triplet Networks

This repository contains a PyTorch implementation for triplet networks.

The code provides two different ways to load triplets for the network. First, it contain a simple MNIST Loader that generates triplets from the MNIST class labels. Second, this repository provides a Triplet Loader that loads images from folders, provided a list of triplets.

Example usage:

$ python train.py

Tracking experiments with Visdom

This repository allows to track experiments with visdom. You can use the VisdomLinePlotter to plot training progress.

If this implementation is useful to you and your project, please also consider to cite or acknowledge this code repository.

About

CS231N: Project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •