• Stars
    star
    505
  • Rank 87,373 (Top 2 %)
  • Language
    Python
  • Created almost 9 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

🚀CNN-based image retrieval built on Keras

Image Retrieval Engine Based on Keras

License

演示

演示地址(不能访问,没钱续VPS了),跑在CPU上,web界面采用的SoTu

环境

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 <path-to-dataset> -index <name-for-output-index>

  • 步骤二

python query_online.py -query <path-to-query-image> -index <path-to-index-flie> -result <path-to-images-for-retrieval>

├── database 图像数据集
├── extract_cnn_vgg16_keras.py 使用预训练vgg16模型提取图像特征
|── index.py 对图像集提取特征,建立索引
├── query_online.py 库内搜索
└── README.md

示例

# 对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

问题汇总

  • query_online.py line 28报错,将index.py line 62注释,使用line 61.

此外,若在使用过程中出现任何问题,请在赞赏留言中留下您的微信,收到消息后我会尽快修复。

A small reward is highly appreciated! (#^.^#) Thank you~

More Repositories

1

awesome-cbir-papers

📝Awesome and classical image retrieval papers
1,625
star
2

cnn-for-image-retrieval

🌅The code of post "Image retrieval using MatconvNet and pre-trained imageNet"
MATLAB
631
star
3

hashing-baseline-for-image-retrieval

:octocat:Various hashing methods for image retrieval and serves as the baselines
MATLAB
596
star
4

pcv-book-code

📖Python计算机视觉中译本实例代码
Python
487
star
5

cnn-cbir-benchmark

CNN CBIR benchmark (ongoing)
Python
187
star
6

cvt

CVT, a Computer Vision Toolkit.
C
164
star
7

image-retrieval

image retrieval
C++
102
star
8

CNN-for-Face-Image-Retrieval

🍎Face image rerieval
MATLAB
90
star
9

practical-python-and-opencv

🍏Latex: translate the book practical python and opencv to Chinese
TeX
87
star
10

opencv-practical-code

🍎OpenCV小项目与OpenCV一些非常有用的代码
C++
74
star
11

SeetaFaceLib

🍑 SeetaFace for Xcode and Qt
C++
74
star
12

willard-yuan.github.io

💎 Yong's Homepage
CSS
67
star
13

py-cbir-image-search-engine

This is a image search engine based on BoW model
Python
47
star
14

my-awesome-awesomeness

自己使用的一些工具
22
star
15

practical-cbir-handbook

A book tries to give some guide for content-based image retrieval
TeX
19
star
16

pupil-vision

C++
17
star
17

video-text-retrieval-papers

15
star
18

tf-dml-cbir

Deep metric learning for CBIR, based on triplet-reid
Python
11
star
19

cv-and-resume

☕My resume for a job
TeX
11
star
20

sparse-reconstruction-hashing

The code of "Learning Hash Functions Using Sparse Reconstruction"
MATLAB
9
star
21

pcvwithpython

Programming Computer Vision with Python中译本
CSS
8
star
22

BoVW

A c++ implemention of bag of virtual words
C++
8
star
23

sift-match-with-ransac

C++
8
star
24

pkbigdata-image-search

:octocat:15万搜索结果
5
star
25

blcf

Python
5
star
26

qt-learning

Notes and codes to learn Qt
C++
4
star
27

tf_cpp

Tensorflow Cpp API Application
C++
4
star
28

awesome-neural-search-papers

4
star
29

cbirbook

4
star
30

machine-learning-notebook

📓My machine learning notebook
Jupyter Notebook
4
star
31

pic

🌐 🍅 PicSearch: CNN web demo for image retrieval
CSS
3
star
32

mykit

Computer Vision ToolKit
Python
3
star
33

mvs

Python
3
star
34

dive-into-caffe

lost in caffe
3
star
35

image-retrieval-based-on-Zernike

MATLAB
2
star
36

cvtk-pypi

C++
2
star
37

slides

tech slides
2
star
38

awesome-text-match-papers

2
star
39

flowingcash

1
star
40

awesome-website-design

收集的设计得很优秀的UI
1
star
41

willard-yuan

1
star
42

libpupil

1
star
43

easyIDCards

C++
1
star
44

mocap

Real-time Human Motion Tracking from Sensors
1
star
45

grpc_toolkit

C++
1
star
46

dlib-practice

C++
1
star
47

matlab-utils

useful utils for myself
MATLAB
1
star
48

habir

Project site of various hashing methods for image retrieval and serves as the baselines
CSS
1
star
49

mt_ins

Python
1
star
50

cpp-code-training

coding
HTML
1
star
51

zh-case-studies

Case Studies 中译本
CSS
1
star