forked from zllrunning/SiameseX.PyTorch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
2f3297e
commit d8454fe
Showing
3 changed files
with
100 additions
and
5 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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} | ||
} | ||
``` | ||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.