Skip to content

Latest commit

 

History

History
47 lines (42 loc) · 1.52 KB

README.md

File metadata and controls

47 lines (42 loc) · 1.52 KB

A pytorch implementation for paper "Feature Learning based Deep Supervised Hashing with Pairwise Labels"

1. Running example:

Environment: python 3

Requirements:

pytorch
torchvision

2. Statement:

As pytorch doesn't provide pretrained VGG-F model, unlike original DPSH paper, we use pretrained Alexnet or pretrained VGG-11 for feature learning part instead of pretrained VGG-F.

3. Data processing:

Following DPSH MatConvNet source code, we can obtain cifar-10.mat. To prepare data for pytorch version DPSH, run script ./data/CIFAR-10/SaveFig.m to save image files.

6. Demo:

python DPSH_CIFAR_10_demo.py

5. Result:

Mean Average Precision on CIFAR-10.

Net StructurePlatForm Code Length
12 bits24 bits 32 bits48 bits
VGG-FMatConvNet 0.713 0.727 0.744 0.757
AlexnetPytorch 0.7505 0.7724 0.7758 0.7828
VGG-11Pytorch 0.7655 0.8042 0.8070 0.8108

Training Loss on CIFAR-10.

6. Influence of Hyper-Parameter \lambda