cwzat / SiamRPN-PyTorch

Implementation SiamRPN on PyTorch with GOT-10k dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SiamRPN-PyTorch

Implementation SiamRPN on PyTorch with GOT-10k dataset

How to run Training

  1. Download the GOT-10k dataset in http://got-10k.aitestunion.com/downloads
  2. Run the train_siamrpn.py script:
cd train

python3 train_siamrpn.py --train_path=/path/to/dataset/GOT-10k/train

How to run Tracking

[Coming Soon]

pip install

pip3 install shapely

How to fix GOT-10k dataset

  1. First you need to delete four videos:
GOT-10k_Train_008628
GOT-10k_Train_008630
GOT-10k_Train_009058  
GOT-10k_Train_009059

Because they are ymin and xmin is greater than the size of the image.

  1. Run the fixed.py script:
python3 fixed.py --dataset_path=/path/to/dataset/GOT-10k/train

After you have new_file.txt file. In this file a lot of information about where the error.

You do not need to change anything yourself, the fixed.py script will do it for you.

My contacts

E-mail: arbi.tularov@yandex.ru

WeChat: tularov_arbi

Authors

Citation

Paper: @InProceedings{Li_2018_CVPR,
author = {Li, Bo and Yan, Junjie and Wu, Wei and Zhu, Zheng and Hu, Xiaolin},
title = {High Performance Visual Tracking With Siamese Region Proposal Network},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2018}
}

About

Implementation SiamRPN on PyTorch with GOT-10k dataset


Languages

Language:Python 100.0%