sanku-lib / image_triplet_loss

Image similarity using Triplet Loss

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

2. To train

python train_triplets.py 

Prediction

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

Prediction.ipynb