Prerequisites
- Requirements for Caffe, pycaffe and matcaffe.
- 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