/image_triplet_loss

Image similarity using Triplet Loss

Primary LanguageJupyter Notebook

Triplet Loss for Image Similarity using tensorflow

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

Facebook FAISS is used for calculating similarity

Requirements

  • Python 3
  • Pip 3
  • Tensorflow (1.13.1)
  • Matplotlib
  • Requests

Environment Setup

Execute requirements.txt to install dependency packages

pip install -r requirements.txt

Training

python train_triplets.py

Prediction

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

Prediction.ipynb