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.
30 lines
641 B
30 lines
641 B
# pytorch_deephash
|
|
|
|
## Introduction
|
|
|
|
This is the Pytorch implementation of [Deep Learning of Binary Hash Codes for Fast Image Retrieval](https://github.com/kevinlin311tw/caffe-cvprw15), and can achieve more than 93% mAP in CIFAR10 dataset.
|
|
|
|
## Environment
|
|
|
|
> Pytorch 0.4.0
|
|
|
|
> torchvision 0.2.1
|
|
|
|
## Training
|
|
|
|
```python
|
|
python train.py
|
|
```
|
|
|
|
You will get trained models in model folder by default, and models' names are their test accuracy.
|
|
|
|
## Evaluation
|
|
|
|
```shell
|
|
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.
|