/Siamese-RPN-tensorflow

An re-implementation for Siamese-RPN with Tensorflow

Primary LanguagePythonMIT LicenseMIT

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