https://github.com/willard-yuan/flask-keras-cnn-image-retrieval
🚀CNN-based image retrieval built on Keras
https://github.com/willard-yuan/flask-keras-cnn-image-retrieval
cbir image-retrieval
Last synced: 2 months ago
JSON representation
🚀CNN-based image retrieval built on Keras
- Host: GitHub
- URL: https://github.com/willard-yuan/flask-keras-cnn-image-retrieval
- Owner: willard-yuan
- Created: 2016-01-12T16:01:16.000Z (over 9 years ago)
- Default Branch: master
- Last Pushed: 2022-04-16T14:45:20.000Z (about 3 years ago)
- Last Synced: 2025-03-28T21:06:42.359Z (2 months ago)
- Topics: cbir, image-retrieval
- Language: Python
- Homepage:
- Size: 165 KB
- Stars: 520
- Watchers: 28
- Forks: 175
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Image Retrieval Engine Based on Keras
[](../LICENSE)
## 演示
[演示地址](http://202.120.39.161:55555/)(不能访问，没钱续VPS了)，跑在CPU上，web界面采用的[SoTu](https://github.com/willard-yuan/SoTu)
## 环境
```python
In [1]: import keras
Using Theano backend.
```
keras 2.0.1 及 2.0.5 版本均经过测试可用。推荐Python 2.7，支持Python 3.6.
此外需要numpy, matplotlib, os, h5py, argparse. 推荐使用anaconda安装
### 使用
- 步骤一
`python index.py -database -index `
- 步骤二
`python query_online.py -query -index -result `
```sh
├── database 图像数据集
├── extract_cnn_vgg16_keras.py 使用预训练vgg16模型提取图像特征
|── index.py 对图像集提取特征，建立索引
├── query_online.py 库内搜索
└── README.md
```
#### 示例
```sh
# 对database文件夹内图片进行特征提取，建立索引文件featureCNN.h5
python index.py -database database -index featureCNN.h5
# 使用database文件夹内001_accordion_image_0001.jpg作为测试图片，在database内以featureCNN.h5进行近似图片查找，并显示最近似的3张图片
python query_online.py -query database/001_accordion_image_0001.jpg -index featureCNN.h5 -result database
```
### 论文推荐
[**awesome-cbir-papers**](https://github.com/willard-yuan/awesome-cbir-papers)
### 问题汇总
- `query_online.py` line 28报错，将`index.py` line 62注释，使用line 61.
此外，若在使用过程中出现任何问题，请在赞赏留言中留下您的微信，收到消息后我会尽快修复。
