FSMT

Installation

conda create -n fsmt python=3.8
conda activate fsmt
pip install -r requirements.txt

Data

Set dataset paths in:

lib/train/admin/local.py
lib/test/evaluation/local.py

Default config:

experiments/fsmt/fsmt_b256.yaml

<hsi_dir>/
|-- list.txt
`-- <sequence_name>/
    |-- HSI/
    |   |-- 0001.png
    |   `-- groundtruth_rect.txt
    |-- HSI-FalseColor/
    |   `-- 0001.jpg
    |-- full_occlusion.txt
    `-- out_of_view.txt

Training

python tracking/train.py --script fsmt --config fsmt_b256 --save_dir ./output --mode single

For multi-GPU training:

python tracking/train.py --script fsmt --config fsmt_b256 --save_dir ./output --mode multiple --nproc_per_node 2

Testing

Set the checkpoint path in:

lib/test/parameter/fsmt.py

python tracking/test.py --tracker_name fsmt --tracker_param fsmt_b256 --dataset_name hsiandfrgb --num_gpus 1 --threads 0