We fork the repository from Caffe and make our modifications.
- caffe
- python 2.7
You can download the data set and labels from UCMerced-4. The password is ti2v. The train label file and test label file is train_label.txt and test_label.txt, respectively.
After download the data set, you will change the data set path in label files.
For example, the path in my train label file is /home/lrh/dataset/UCdataset-4/agricultural00.jpg 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 where 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 is agricultural00.jpg's label, and you need to replace the path of the data set with your path your path/UCdataset-4/agricultural00.jpg 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0.
You can download the pretrained model here. The password is d4uo. You need to put the trained model ResUCMD32.caffemodel in ./models/Resnet-50/.
In ./models/ResNet-50/predict/, we give a test python file predict_parallel.py to show how to evaluate the trained hash model.
python predict_parallel.py