An image search implementation in python using tensorflow keras, scikit-learn, scipy and matplotlib.
The image embeddings are generated from Xception imagenet (can be changed/tuned from features.py). Currently the embeddings are stored using pickle, but a database may be used instead. Image embeddings are compared cosine similarity and hamming.
- Run search.py with image path(s) as arguments and the script will display the top matches from underlying image library
- Transfer the images to queries folder data/queries/ or alternatively change
query_images_folder_path
from paths.py to your location of your image set. - Run search.py and the script will display the top matches from underlying image library
- Transfer the images to images folder data/images/ or alternatively change
images_folder_path
from paths.py to your location of your image set. - Run features.py and you should have the embeddings generated
- Run dataset.py and you should get a visualization similar to this
The sample images used here are from idenprof dataset