Skip to content

Latest commit

 

History

History
20 lines (15 loc) · 719 Bytes

README.md

File metadata and controls

20 lines (15 loc) · 719 Bytes

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.