/sis

Simple image search engine

Primary LanguagePythonApache License 2.0Apache-2.0

Language: 🇺🇸 🇨🇳

«sis» provides a simple image search engine based on Pytorch/Flask/Docker

Table of Contents

Background

This warehouse is developed based on matsui528/sis, provides image retrieval function based on pytorch, and provides more stable functions combined with flask+docker.

Installation

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

Usage

Local deployment

  • First, you should put gallery images to ./static/img
  • Then, extract features using offline.py
python offline.py
  • Finally, launch flask app like this:
gunicorn server:app -c gunicorn.conf.py

Container deployment

  • First, build an image
docker build -t sis:0.1.0 .
  • Second, run it
bash bash_docker.sh

Maintainers

  • Yusuke Matsui - Initial work - matsui528
  • zhujian - Enhanced work - zjykzj

Thanks

Contributing

Anyone's participation is welcome! Open an issue or submit PRs.

Small note:

License

Apache License 2.0 © 2022 zjykzj