Skip to content

smallsmallflypigtang/Deep-Hash-learning-for-Remote-Sensing-Image-Retrieval

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep-Hash-learning-for-Remote-Sensing-Image-Retrieval

The official codes for paper "Deep hash learning for remote sensing image retrieval"

Install dependencies

numpy
opencv-python
torch
torchvision

Data

We conduct the experiments on three data sets, including UC Merced, AID, and NWPU-RESISC45. To train and test our model, you should download the data set and modify each image's path in the dataset/AID/.txt or dataset/NWPU/.txt or dataset/UC_Merced/.txt (depending which data set you select to conduct the experiment)

Training

All the configurations are in trainerAndHash.py, and you can modify them by your needs.

train the model

python trainerAndHash.py --phase=0

extract hash codes

python trainerAndHash.py --phase=1

the path of codes can be modified the line about "parser.add_argument('--codes_dir', default=root + '/codes', type=str)" in trainerAndHash.py

calculate the (mean average precision) mAP value

python trainerAndHash.py --phase=2

About

The official codes for paper "Deep hash learning for remote sensing image retrieval"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages