pip install -r requirements.txt
- pytorch >= 1.0
- loguru
- cifar10-gist.mat password: umb6
- cifar-10_alexnet.t password: f1b7
- nus-wide-tc21_alexnet.t password: vfeu
- imagenet-tc100_alexnet.t password: 6w5i
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)
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 |