pytorch_deephash

Introduction

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.

Environment

Pytorch 0.4.0

torchvision 0.2.1

Training

python train.py

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

Evaluation

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

Tips

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