This is the Pytorch implementation of Deep Learning of Binary Hash Codes for Fast Image Retrieval, and can achieve more than 93% mAP in CIFAR10 dataset.

Pytorch 0.4.0

torchvision 0.2.1

python train.py

You will get trained models in model folder by default, and models' names are their test accuracy.

python mAP.py --pretrained {your saved model name in model folder by default}

There are some other args, which you can get them by adding '-h' or reading the code.