Skip to content

Latest commit

 

History

History
35 lines (28 loc) · 778 Bytes

README.md

File metadata and controls

35 lines (28 loc) · 778 Bytes

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