Skip to content

sanku-lib/image_triplet_loss

Repository files navigation

Triplet Loss for Image Similarity using tensorflow

This repository is an implementation of following "medium" story: Image similarity using Triplet Loss

Requirements

  • Python 3
  • Pip 3
  • Tensorflow
  • Matplotlib
  • Requests

Environment Setup

Execute requirements.txt to install dependency packages

pip install -r requirements.txt

Training

  1. Download Training Dataset by executing download_dataset.py
python download_dataset.py
  1. To train
python train_triplets.py 

Prediction

Run Prediction.ipynb using Jupyter notebook to look into Prediction code.

Prediction.ipynb