Re-trained VGG16 Model on Caltech Image dataset with around 9,000 images for training, using Keras, for searching and retrieving images similar to the given query image.
Download the dataset from (http://www.vision.caltech.edu/Image_Datasets/Caltech101/)
Used libraries: -
- Keras
- PCA
- SciPy
- Tqdm
- Matplotlib
Tensorflow backend is used.
Steps involved: -
- Imported VGG16 and modified it to extract features from images.
- PCA is used to reduce the number of required features per image.
- Euclidean Distance is used to define the relation between images. Each pair of images with small Euclidean distance are considered to be similar.
- Tqdm is a library to visualize a loop in the form of a progressive bar.
- Query image and the resulting related images are shown using Matplotlib.