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 1.4.0

torchvision 0.5.0

tqdm

numpy

python train.py

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

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

1. If using Windows, keep num_works zero

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