A Barebones Image Retrieval System
- 0 Collaborators
This project presents a simple framework to retrieve images similar to a query image. ...learn more
Project status: Under Development
Intel Technologies
Intel Python
Overview / Usage
This project presents a simple framework to retrieve images similar to a query image.
Methodology / Approach
This project presents a simple framework to retrieve images similar to a query image. The framework is as follows:
- Train a CNN model (A) on a set of labeled images with Triplet Loss (I used this one).
- Use the trained CNN model (A) to extract features from the validation set.
- Train a kNN model (B) on these extracted features with k set to the number of neighbors wanted.
- Grab an image (I) from the validation set and extract its features using the same CNN model (A).
- Use the same kNN model (B) to calculate the nearest neighbors of I.
Technologies Used
TensorFlow
Repository
https://github.com/sayakpaul/A-Barebones-Image-Retrieval-System