Skip to content

The Deep Supervised Hashing for Image Retrieval on CIFAR10/MNIST/Fashion-MNIST

Notifications You must be signed in to change notification settings

mingloo/DeepSupervisedHashing

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Supervised Hashing for Image Retrieval

This is non-official Torch implementation with minor modification for Deep Supervised Hashing for Fast Image Retrieval.

The official Caffe implementation can be found here.

Prerequisites

  • Install Torch

  • Datasets

    • Install MNIST: luarocks install mnist
    • Install Fashion-MNIST: luarocks install https://raw.github.com/mingloo/fashion-mnist/master/rocks/fashion-mnist-scm-1.rockspec
    • CIFAR10: the code could directly download this dataset.
  • matio: we use matlab scripts that adopted from here to calculate retrieval Mean Average Precision, mAP.

  • weight-init.lua is adopted from here.

Usage

Please run the run.sh for training model from scratch and obtaining final hashing representation with default parameter settings for CIFAR10/MNIST/Fashion-MNIST datasets:

bash run.sh

Then copy the output result with suffix .mat in output/ directory to the scripts/ directory and run DSH.m to calculate the retrieval mAP.

Results

DSH-CIFAR10

DSH-MNIST

DSH-Fashion-MNIST

About

The Deep Supervised Hashing for Image Retrieval on CIFAR10/MNIST/Fashion-MNIST

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published