/image_retrieval_platform

This is an simple image retrieval platform by CNN based on pytorch and flask.

Primary LanguagePython

A simple image retrieval demo with pytorch and flask

This is a simple demo of image retrieval based on pretrained CNN.

Demo.

The demo video is shown downside. image

Usage.

Create folder "models" in "retrieval" folder, and "thumb_images" in "static/" folder. Get the pretrained CNN model from this link and put the model in path "./retrieval/models/".

run the following command:

$ python image_retrieval_cnn.py

File orgnization.

If you only want to test the retrieval proccess, just read the code image_retrieval_cnn.py for reference.