Skip to content

rayyao/HDSP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Hyperspectral Object Tracking with Dual-Stream Prompt

Installation

Create and activate a conda environment:

conda create -n hdsp python=3.7
conda activate hdsp
Install the required packages
bash install_hdsp.sh

Path Setting

Run the following command to set paths cd <PATH_of_HDSP>

python tracking/create_default_local_file.py --workspace_dir . --data_dir ./data --save_dir ./output

Training

Download the pretrained foundation model (OSTrack) and put it under ./pretrained/.

python tracking/train.py --script HDSP --config hdsp_deep --save_dir ./output --mode multiple --nproc_per_node 2

###Testing Modify the <DATASET_PATH> and <SAVE_PATH>

python HSI_test/tsetHSI.py  --script_name HDSP --dataset_name HSItest --yaml_name hdsp_deep

Citation

If these codes are helpful for you, please cite this paper:

BibTex Format:

@ARTICLE{10798510,
 author={Yao, Rui and Zhang, Lu and Zhou, Yong and Zhu, Hancheng and Zhao, Jiaqi and Shao, Zhiwen},
 journal={IEEE Transactions on Geoscience and Remote Sensing}, 
 title={Hyperspectral Object Tracking With Dual-Stream Prompt}, 
 year={2025},
 volume={63},
 number={5500612},
 pages={1-12}}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published