Skip to content

confessant-1/SiamMSS

 
 

Repository files navigation

SiamMSS

1. Environment setup

This code has been tested on windows 10, Python 3.7.9, Pytorch 1.7.0, CUDA 10.0. Please install related libraries before running this code:

pip install -r requirements.txt

2. Test

Download the pretrained model and put them into tools/snapshot directory.
From BaiduYun:

If you want to test the tracker on a new dataset, please refer to pysot-toolkit to set test_dataset.

The tracking result can be download from BaiduYun (extract code: 50fq) for comparision.

python testTracker.py \    
        --config ../experiments/siamsmm_googlenet/config.yaml \
	--dataset LSOTB \                                 # dataset_name
	--snapshot snapshot/siamsmm.pth              # tracker_name

The testing result will be saved in the results/dataset_name/tracker_name directory.

3. Train

Prepare training datasets

Download the datasets:

Note: training_dataset/dataset_name/readme.md has listed detailed operations about how to generate training datasets.

Download pretrained backbones

Download pretrained backbones from link and put them into pretrained_models directory.

Train a model

To train the SiamSMM model, run train.py with the desired configs:

cd tools
python train.py

5. Acknowledgement

The code is implemented based on pysot and SiamGAT. We would like to express our sincere thanks to the contributors.

Releases

No releases published

Packages

No packages published

Languages

  • Python 82.2%
  • C 9.9%
  • Cython 4.8%
  • C++ 3.0%
  • Makefile 0.1%