Skip to content

makalo/Siamese-RPN-tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Siamese-RPN-tensorflow

Code for reproducing the results in the following paper:

Environment

  • python=3.6
  • tensorflow=1.10
  • cuda=9.0

Downloading VOT2013 Data

Downloading YouTube-bb Data

Downloading ILSVRC 2015-VID Data

Performance

The red box is for tracing

Visualization for debug

bbox in detection

  • red -- the groundtruth

  • black -- bbox with highest score

  • other colors -- bbox with scores from second to tenth.

Training and Evaluation

If your data format is the same as VOT 2013, you can run the code directly. If not, you need to change the utils/image_reader.py or convert the data format to VOT format.

To train Siamese-RPN:

python train.py

If you want to see if the training is reasonable in the course of training, you can choose to turn on debug.Just change the init() in train.py

self.is_debug=True

This will result in a debug folder where you can see pictures of the training process, with groundtruth in red and box in top 10 scores in other colors.

To test Siamese-RPN:

To test series of images like VOT format

If you want to test a series of images captured from the video, you need to assign new values img_pathand img_label in config.py, which are the files of your image's path and label, respectively. Then execute the following commands

python test.py

This command will automatically synthesize videos from image sequences, and also synthesize videos from processed images, which are saved in. / data / vedio

To test a vedio

If you are testing a video, you need to put the video in./data/vedio. You can run the following command and select the object you want to track in the first frame according to the program prompt at the beginning.

python vedio_test.py test.mp4

The 'test.mp4' is the name of your vedio

Model

I will provide the well-trained model in the next few days

About

An re-implementation for Siamese-RPN with Tensorflow

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages