All Projects → willard-yuan → Flask Keras Cnn Image Retrieval

willard-yuan / Flask Keras Cnn Image Retrieval

🚀CNN-based image retrieval built on Keras

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Flask Keras Cnn Image Retrieval

ISeeNN
A CNN feature based image retrieval website
Stars: ✭ 15 (-96.39%)
Mutual labels:  image-retrieval
Trace.moe
Anime Scene Search by Image
Stars: ✭ 3,231 (+676.68%)
Mutual labels:  image-retrieval
Batch Dropblock Network
Official source code of "Batch DropBlock Network for Person Re-identification and Beyond" (ICCV 2019)
Stars: ✭ 304 (-26.92%)
Mutual labels:  image-retrieval
symmetrical-synthesis
Official Tensorflow implementation of "Symmetrical Synthesis for Deep Metric Learning" (AAAI 2020)
Stars: ✭ 67 (-83.89%)
Mutual labels:  image-retrieval
Siamesenetwork Tensorflow
Using siamese network to do dimensionality reduction and similar image retrieval
Stars: ✭ 258 (-37.98%)
Mutual labels:  image-retrieval
Trace.moe Telegram Bot
This Telegram Bot can tell the anime when you send an screenshot to it
Stars: ✭ 284 (-31.73%)
Mutual labels:  image-retrieval
art
Exploring the connections between artworks with deep "Visual Analogies"
Stars: ✭ 73 (-82.45%)
Mutual labels:  image-retrieval
Hardnet
Hardnet descriptor model - "Working hard to know your neighbor's margins: Local descriptor learning loss"
Stars: ✭ 350 (-15.87%)
Mutual labels:  image-retrieval
Powerful Benchmarker
A PyTorch library for benchmarking deep metric learning. It's powerful.
Stars: ✭ 272 (-34.62%)
Mutual labels:  image-retrieval
Person Reid gan
ICCV2017 Unlabeled Samples Generated by GAN Improve the Person Re-identification Baseline in vitro
Stars: ✭ 301 (-27.64%)
Mutual labels:  image-retrieval
kapture-localization
Provide mapping and localization pipelines based on kapture format
Stars: ✭ 111 (-73.32%)
Mutual labels:  image-retrieval
diffusion-retrieval
Efficient Diffusion on Region Manifolds: Recovering Small Objects with Compact CNN Representations
Stars: ✭ 78 (-81.25%)
Mutual labels:  image-retrieval
Deepvideoanalytics
A distributed visual search and visual data analytics platform.
Stars: ✭ 2,973 (+614.66%)
Mutual labels:  image-retrieval
Similarity-Adaptive-Deep-Hashing
Unsupervised Deep Hashing with Similarity-Adaptive and Discrete Optimization (TPAMI2018)
Stars: ✭ 18 (-95.67%)
Mutual labels:  image-retrieval
Artificio
Deep Learning Computer Vision Algorithms for Real-World Use
Stars: ✭ 326 (-21.63%)
Mutual labels:  image-retrieval
gvcnn-tf
Group-View Convolutional Neural Networks for 3D Shape Recognition
Stars: ✭ 22 (-94.71%)
Mutual labels:  image-retrieval
Pytorch Metric Learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Stars: ✭ 3,936 (+846.15%)
Mutual labels:  image-retrieval
Cbir
🏞 A content-based image retrieval (CBIR) system
Stars: ✭ 407 (-2.16%)
Mutual labels:  image-retrieval
Mmt
[ICLR-2020] Mutual Mean-Teaching: Pseudo Label Refinery for Unsupervised Domain Adaptation on Person Re-identification.
Stars: ✭ 345 (-17.07%)
Mutual labels:  image-retrieval
Computer Vision Leaderboard
Comparison of famous convolutional neural network models
Stars: ✭ 299 (-28.12%)
Mutual labels:  image-retrieval

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

更新

  • 针对近期有小伙伴反映的keras版本的问题,已将其进行到最新版本,并且特征提取代码大幅精简。
  • 显示检索得到的图片, 可自由定义查询图片及检索图片集

Goal

  • [x] 重新用flask写web界面,已完成。

论文推荐

awesome-cbir-papers

问题汇总

  • query_online.py line 28报错,将index.py line 62注释,使用line 61.
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].