arbitularov / SiamRPN-PyTorch

Implementation SiamRPN on PyTorch with GOT-10k dataset
103 stars 20 forks source link
got-10k object-tracking pytorch siamese-network siamrpn tracker

I got on OTB2015 result 8.41 and 0.625 without RPN. In the SiamFusion project

result

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

<center>
    <figure>
        <img src="https://github.com/arbitularov/SiamRPN-PyTorch/raw/master/img/error.png" height="60%" width="100%">
        <figcaption>
        </figcaption>
    </figure>
</center>

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.

2. Run the fixed.py script:

python3 fixed.py --dataset_path=/path/to/dataset/GOT-10k/train

<center>
    <figure>
        <img src="https://github.com/arbitularov/SiamRPN-PyTorch/raw/master/img/100.png" height="60%" width="100%">
        <figcaption>
        </figcaption>
    </figure>
</center>

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

<center>
    <figure>
        <img src="https://github.com/arbitularov/SiamRPN-PyTorch/raw/master/img/new_file.png" height="60%" width="60%">
        <figcaption>
        </figcaption>
    </figure>
</center>

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

<center>
    <figure>
        <img src="https://github.com/arbitularov/SiamRPN-PyTorch/raw/master/img/WeChat.jpg" height="30%" width="30%">
        <figcaption>
        </figcaption>
    </figure>
</center>

## Authors

* `Bo Li` - paper - [Siamese-RPN](http://openaccess.thecvf.com/content_cvpr_2018/papers/Li_High_Performance_Visual_CVPR_2018_paper.pdf)
* `De jiasong` - code - [Siamese-RPN-pytorch](https://github.com/songdejia/Siamese-RPN-pytorch)
* `Makalo` - code - [Siamese-RPN-tensorflow](https://github.com/makalo/Siamese-RPN-tensorflow)

## 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} }