Skip to content

Source code for paper "Spectral Hashing" on NIPS-2009

License

Notifications You must be signed in to change notification settings

TreezzZ/SH_PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Spectral Hashing

REQUIREMENTS

pip install -r requirements.txt

  1. pytorch >= 1.0
  2. loguru

DATASETS

  1. cifar10-gist.mat password: umb6
  2. cifar-10_alexnet.t password: f1b7
  3. nus-wide-tc21_alexnet.t password: vfeu
  4. imagenet-tc100_alexnet.t password: 6w5i

USAGE

usage: run.py [-h] [--dataset DATASET] [--root ROOT]
              [--code-length CODE_LENGTH] [--topk TOPK] [--gpu GPU]
              [--seed SEED]

SH_PyTorch

optional arguments:
  -h, --help            show this help message and exit
  --dataset DATASET     Dataset name.
  --root ROOT           Path of dataset
  --code-length CODE_LENGTH
                        Binary hash code length.(default:
                        8,16,24,32,48,64,96,128)
  --topk TOPK           Calculate map of top k.(default: ALL)
  --gpu GPU             Using gpu.(default: False)
  --seed SEED           Random seed.(default: 3367)

EXPERIMENTS

cifar10-gist dataset. Gist features, 1000 query images, 5000 training images, MAP@ALL.

cifar-10-alexnet dataset. Alexnet features, 1000 query images, 5000 training images, MAP@ALL.

nus-wide-tc21-alexnet dataset. Alexnet features, top 21 classes, 2100 query images, 10500 training images, MAP@5000.

imagenet-tc100-alexnet dataset. Alexnet features, top 100 classes, 5000 query images, 10000 training images, MAP@1000.

Bits 8 16 24 32 48 64 96 128
cifar10-gist@ALL 0.1246 0.1303 0.1290 0.1283 0.1310 0.1286 0.1288 0.1274
cifar10-alexnet@ALL 0.1890 0.1863 0.1912 0.1892 0.2044 0.2013 0.1978 0.1960
nus-wide-tc21-alexnet@5000 0.5503 0.5615 0.5886 0.6402 0.6309 0.6350 0.6344 0.6411
imagenet-tc100-alexnet@1000 0.0961 0.2013 0.2421 0.2806 0.3235 0.3445 0.3747 0.3908

About

Source code for paper "Spectral Hashing" on NIPS-2009

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages