Skip to content

Commit

Permalink
Fixed README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
Jiwoong Choi committed Sep 16, 2019
1 parent 9a5a3d7 commit 41b6e16
Showing 1 changed file with 14 additions and 4 deletions.
18 changes: 14 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,22 @@
You can use the models (with the refine module) trained with the original repository [foolwood/SiamMask](https://github.com/foolwood/SiamMask) for inference in C++. Just Follow the instruction in [jiwoong-choi/SiamMask](https://github.com/jiwoong-choi/SiamMask#converting-siammask-model-with-the-refine-module-to-torch-scripts) to convert your own models to Torch script files.

# Download pretrained Torch scripts
Or you can download pretrained Torch scripts from the [release page](https://github.com/nearthlab/SiamMaskCpp/releases).
These are converted from the pretrained models ([SiamMask_DAVIS.pth](http://www.robots.ox.ac.uk/~qwang/SiamMask_DAVIS.pth) and [SiamMask_VOT.pth](http://www.robots.ox.ac.uk/~qwang/SiamMask_VOT.pth)) in the original repository.
Or you can download pretrained Torch scripts. These files are converted from the pretrained models ([SiamMask_DAVIS.pth](http://www.robots.ox.ac.uk/~qwang/SiamMask_DAVIS.pth) and [SiamMask_VOT.pth](http://www.robots.ox.ac.uk/~qwang/SiamMask_VOT.pth)) in the original repository.

# How to build demo
```bash
git clone --recurse-submodules https://github.com/nearthlab/SiamMaskCpp
cd SiamMaskCpp
mkdir models
cd models
wget https://github.com/nearthlab/SiamMaskCpp/releases/download/v1.0/SiamMask_DAVIS.tar.gz
wget https://github.com/nearthlab/SiamMaskCpp/releases/download/v1.0/SiamMask_VOT.tar.gz
tar -xvzf SiamMask_DAVIS.tar.gz
tar -xvzf SiamMask_VOT.tar.gz
```

# How to build demo
```bash
cd SiamMaskCpp
mkdir build
cd build
cmake -DCMAKE_PREFIX_PATH=/path/to/python3.x/site-packages/torch ..
Expand All @@ -23,5 +32,6 @@ make
# How to run demo
```bash
cd SiamMaskCpp/build
./demo -c ../config_davis.cfg -m ../SiamMask_DAVIS ../tennis
./demo -c ../config_davis.cfg -m ../models/SiamMask_DAVIS ../tennis
./demo -c ../config_vot.cfg -m ../models/SiamMask_VOT ../tennis
```

0 comments on commit 41b6e16

Please sign in to comment.