You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
641 B
641 B
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.