Skip to content

CNN-based image retrieval with TensorFlow and Keras

Notifications You must be signed in to change notification settings

setezzy/image-retrieval

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 

Repository files navigation

image-retrieval

Deep Learning for content-based image retrieval with TensorFlow and Keras

tensorflow version

usage: python retrieval.py -h

img_trans :

skimage and keras are required in this file. The following transformations are included:

  • adjust brightness
  • shit, rotate, flip, zoom
  • dilation, erosion
  • add oblique line
  • add salt noise

lenet5_model :

LeNet-5 is a classical CNN model proposed by Yann LeCun.

See Gradient-based learning applied to document recognition for more details.

ReLu is used as the activate function for convlayer and fclayer. Drop-out layer is added before softmax in this implementation.

lenet_train :

Train lenet-5 on MNIST.

retrieval :

Extract features of query image and all retrieval images from fully-connected layer.

Perform feature similarity computation for retrieval task. Cosine similarity is adopted in this implementation.

retrieval_ :

Example of using images stored on disk as tf inputs.

keras version

utils : Image preprocessing.

funcs : Image retrieval function.

lenet5 : Implementation of Lenet-5 with keras.

About

CNN-based image retrieval with TensorFlow and Keras

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages