Skip to content

Commit

Permalink
update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
zllrunning committed Apr 21, 2019
1 parent 2f3297e commit d8454fe
Show file tree
Hide file tree
Showing 3 changed files with 100 additions and 5 deletions.
8 changes: 5 additions & 3 deletions .idea/workspace.xml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

97 changes: 95 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,7 +1,100 @@
# SiameseX.PyTorch
A simplified PyTorch implementation of Siamese networks for tracking: SiamFC, SiamVGG, SiamDW.
A simplified PyTorch implementation of Siamese networks for tracking: SiamFC, SiamVGG, SiamDW.

**Warning: It is still in development.**

## Dependencies
- python2.7
- pytorch == 0.4.0
- opencv
- opencv

## Models
- [SiamFC](https://arxiv.org/abs/1606.09549)
- [SiamVGG](https://arxiv.org/abs/1902.02804)
- [SiamFCRes22](https://arxiv.org/abs/1901.01660)
- [SiamFCIncep22](https://arxiv.org/abs/1901.01660)
- [SiamFCNext22](https://arxiv.org/abs/1901.01660)

## Backbones
- AlexNet
- VGG
- ResNet22
- Incep22
- ResNeXt22

## Demo
- Clone this repo and run
```
python demo.py --model SiamFCNext22
```

You can change `--mdoel` to other models like
```
python demo.py --model SiamFC
```

- You'll see the following:
<div align="center">
<img src="data/bag.gif" width="400px" />
</div>

## Training

- download [VID dataset](http://bvisionweb1.cs.unc.edu/ilsvrc2015/download-videos-3j16.php) (I did not use Youtube-bb because of limited resources).
- run `label_preprocess.py --file --output_file --path` to replace my file path by yours, **these three arguments should be treated carefully **.
- train SiamFCNext22 tracker
```
python train.py --model SiamFCNext22 --gpu 0 --task siamresnext
```
- just replace `--model` argument and you can train other models
```
python train.py --model SiamVGG --gpu 0 --task siamvgg
```

## TODO
We have accumulated the following to-do list, which we hope to complete in the near future
- Still to come:
* [ ] Add testing code on common datasets
* [ ] Add SiamRPN(AlexNet as backbone)
* [ ] Add SiamRPN(VGG as backbone)
* [ ] Add SiamRPN(ResNet, ResNext, Inception as backbone)

### Citation

```
@inproceedings{bertinetto2016fully,
title={Fully-convolutional siamese networks for object tracking},
author={Bertinetto, Luca and Valmadre, Jack and Henriques, Joao F and Vedaldi, Andrea and Torr, Philip HS},
booktitle={European conference on computer vision},
pages={850--865},
year={2016},
organization={Springer}
}
@inproceedings{Li2019SiamVGGVT,
title={SiamVGG: Visual Tracking using Deeper Siamese Networks},
author={Yuhong Li and Xiaofan Zhang},
year={2019}
}
@inproceedings{SiamDW_2019_CVPR,
author={Zhang, Zhipeng and Peng, Houwen},
title={Deeper and Wider Siamese Networks for Real-Time Visual Tracking},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year = {2019}
}
```














Binary file added data/bag.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit d8454fe

Please sign in to comment.