https://github.com/ankitsharma-tech/image-triplet-loss
Image similarity using Triplet Loss.
https://github.com/ankitsharma-tech/image-triplet-loss
cnn-tensorflow deep-learning image-processing image-similarity machine-learning siamese-network tensorflow triplet-loss
Last synced: 8 months ago
JSON representation
Image similarity using Triplet Loss.
- Host: GitHub
- URL: https://github.com/ankitsharma-tech/image-triplet-loss
- Owner: ankitsharma-tech
- License: mit
- Created: 2025-09-30T18:14:15.000Z (8 months ago)
- Default Branch: main
- Last Pushed: 2025-09-30T18:15:48.000Z (8 months ago)
- Last Synced: 2025-09-30T20:14:08.818Z (8 months ago)
- Topics: cnn-tensorflow, deep-learning, image-processing, image-similarity, machine-learning, siamese-network, tensorflow, triplet-loss
- Language: Jupyter Notebook
- Homepage:
- Size: 2.94 MB
- Stars: 9
- Watchers: 0
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Triplet Loss for Image Similarity using tensorflow
This repository is an implementation of following "medium" story:
**[Image similarity using Triplet Loss](https://towardsdatascience.com/image-similarity-using-triplet-loss-3744c0f67973?source=friends_link&sk=ad2d8e0921e7cdaec1e65e6e0474df10)**
## Requirements
- Python 3
- Pip 3
- Tensorflow
- Matplotlib
- Requests
## Environment Setup
Execute requirements.txt to install dependency packages
```bash
pip install -r requirements.txt
```
## Training
1. Download Training Dataset by executing download_dataset.py
```bash
python download_dataset.py
```
2. To train
```bash
python train_triplets.py
```
## Prediction
Run Prediction.ipynb using Jupyter notebook to look into Prediction code.
```
Prediction.ipynb
```