darrenzzhang / tcsvt21-dstdh

DSTDH Dual-level Semantic Transfer Deep Hashing

H. Cui, L. Zhu, Z. Cheng, J. Li, Z. Zhang, Dual-level Semantic Transfer Deep Hashing for Efficient Social Image Retrieval, IEEE Transactions on Circuits and Systems for Video Technology (TCSVT), 31(4): 1478-1489, 2021.

Prerequisites

1. Requirements for Caffe, pycaffe and matcaffe.
2. VGG-16 pre-trained model on ILSVC12 datasets, and save it in caffemodels directory.

Installation

Enter caffe directory and download the source codes.

cd caffe/

Modify Makefile.config and build Caffe with following commands:

make all -j8

make pycaffe

make matcaffe

Usage

We only supply the code to train 32-bit DSTDH on MIR Flickr dataset.

We integrate train step and test step in a bash file train32.sh, please run it as follows:

sudo./train32.sh [ROOT_FOLDER] [GPU_ID]

# ROOT_FOLDER is the root folder of image datasets,

# GPU_ID is the GPU you want to train on,

# e.g. sudo ./train32.sh ./flickr_25 1