Source-linked AI summary
Fast Online Object Tracking and Segmentation: A Unifying Approach
Qiang Wang, Li Zhang, Luca Bertinetto, Weiming Hu, Philip H. S. Torr
TL;DR
Visual tracking is efficient but commonly represents objects with coarse bounding boxes, while video object segmentation supplies pixel-level masks but has traditionally been too slow for real-time use. SiamMask combines offline-trained Siamese tracking with multi-task bounding-box regression and binary segmentation. It runs online from one bounding-box initialization at 55 frames per second, achieves state-of-the-art real-time tracking on VOT-2018, and remains competitive while fastest by a large margin on DAVIS benchmarks.
Problem
Video object segmentation offers detailed pixel-level representations but has traditionally been too slow for real-time operation, while tracking commonly uses less precise bounding boxes.
Method
SiamMask jointly trains a fully-convolutional Siamese network for similarity matching, bounding-box regression, and class-agnostic binary segmentation.
Results
55 frames per second: SiamMask produces masks and rotated bounding boxes from one bounding-box initialization, establishes a real-time tracking state of the art on VOT-2018, and is competitive and fastest by a large margin on DAVIS-2016 and DAVIS-2017.
Takeaways & Limitations
A single simple method can perform online visual tracking and semi-supervised video object segmentation in real time without test-time adaptation.
Takeaways & Limitations
SiamMask fails in motion-blur and non-object-instance scenarios associated with a lack of similar training samples.
Abstract
from arXiv · showhide
In this paper we illustrate how to perform both visual object tracking and semi-supervised video object segmentation, in real-time, with a single simple approach. Our method, dubbed SiamMask, improves the offline training procedure of popular fully-convolutional Siamese approaches for object tracking by augmenting their loss with a binary segmentation task. Once trained, SiamMask solely relies on a single bounding box initialisation and operates online, producing class-agnostic object segmentation masks and rotated bounding boxes at 55 frames per second. Despite its simplicity, versatility and fast speed, our strategy allows us to establish a new state of the art among real-time trackers on VOT-2018, while at the same time demonstrating competitive performance and the best speed for the semi-supervised video object segmentation task on DAVIS-2016 and DAVIS-2017. The project website is http://www.robots.ox.ac.uk/~qwang/SiamMask.
1. Introduction
Visual tracking offers efficient online localization from a bounding-box initialization, whereas video object segmentation provides more detailed masks but has traditionally lacked real-time speed. SiamMask narrows this gap by combining tracking and segmentation in one offline-trained, online method.
- Motivation: Visual tracking estimates an arbitrary target’s position across subsequent video frames from its first-frame location.This supports applications including surveillance, navigation, video labelling, human-computer interaction, and activity recognition.
- Motivation: Online trackers process streaming video without using future frames, typically representing targets with axis-aligned or rotated bounding boxes.Bounding-box annotation keeps labelling and target initialization quick and simple.
- Motivation: Video object segmentation instead predicts a binary mask indicating whether each pixel belongs to the target, which is useful for pixel-level applications.Pixel-level estimates are more computationally demanding than bounding boxes.
- Problem: VOS methods have traditionally required several seconds per frame, and even recent faster approaches have not operated in real time.This speed gap motivates a method that combines tracking convenience with segmentation detail.
- Contribution: SiamMask uses a simple multi-task approach to address visual tracking and VOS while retaining offline training, online speed, and a more refined target representation.It augments similarity-based tracking with bounding-box regression and class-agnostic binary segmentation, requiring binary labels only during offline training.
- Contribution: 55 frames per second: SiamMask produces segmentation masks and rotated bounding boxes after a single bounding-box initialization, operates online without updates, and achieves strong results on tracking and VOS benchmarks.It establishes a real-time tracking state of the art on VOT-2018 and is competitive and fastest by a large margin on DAVIS-2016 and DAVIS-2017.
2. Related Work
Prior work separates fast online tracking from more detailed but slower video object segmentation. SiamMask is motivated by combining offline-trained Siamese tracking with binary masks while preserving simple initialization and online operation.
- Visual object tracking: Traditional arbitrary-object trackers commonly train a discriminative classifier online from first-frame ground-truth information and update it during tracking.
- Visual object tracking: Fully-convolutional Siamese trackers instead learn an offline similarity function on video-frame pairs and evaluate it once per frame on new videos.Later variants improve this approach with region proposals, hard-negative mining, ensembling, and memory mechanisms.
- Visual object tracking: Most modern trackers use rectangular bounding boxes for initialization and subsequent localization, although rectangles can poorly represent object shape.This limitation motivates producing binary masks while retaining bounding-box initialization.
- Visual object tracking: Earlier mask-producing trackers either operated slowly, lacked modern benchmark competitiveness, or required test-time fine-tuning.Yeo et al.’s fastest variant ran at 4 frames per second, while CNN features reduced its speed below 0.1 fps.
- Semi-supervised video object segmentation: VOS methods use temporal propagation or independent-frame processing, with examples including graph-labeling methods, optical-flow models, and first-frame fine-tuning.Recent work increasingly targets faster methods, including approaches designed to approach state-of-the-art performance.
- Semi-supervised video object segmentation: VOS methods prioritize accurate object representation and often use computationally intensive techniques such as fine-tuning, data augmentation, or optical flow.These approaches can require minutes or hours for videos lasting only a few seconds.
3. Methodology
SiamMask extends fully-convolutional Siamese trackers with a segmentation branch, enabling pixel-wise masks alongside tracking outputs. It uses shared representations, multi-task losses, and online mask- or box-based reference selection.
- Siamese tracking foundation: SiamMask adopts fully-convolutional Siamese tracking as its starting framework for online operation and fast speed.The method considers SiamFC and SiamRPN as representative starting points.
- Mask prediction: The approach extends Siamese trackers with an extra branch and loss to encode pixel-wise binary masks in each response of a candidate window.It predicts one w×h mask for each candidate window using a two-layer network hφ.
- Mask prediction: Mask prediction depends on both the search image x and exemplar target z, allowing the exemplar to guide segmentation of the search image.A different reference image produces a different segmentation mask for x.
- Training objective: The mask loss is binary logistic regression over candidate windows, with pixel classifiers indicating whether each pixel belongs to the object.Lmask is applied only to positive candidate windows.
- Training objective: SiamMask combines mask learning with similarity, score, and box objectives in two-branch and three-branch variants.The three-branch objective is L3B = λ1 · Lmask + λ2 · Lscore + λ3 · Lbox; training uses λ1 = 32 and λ2 = λ3 = 1.
- Online inference: At inference, the selected mask is thresholded at 0.5, then either fitted with a Min-max box or replaced by the highest-scoring box-branch output for the next search region.The two-branch variant uses the fitted mask box, while the three-branch variant uses the box branch.
4. Experiments
Experiments evaluate SiamMask on visual tracking and semi-supervised video object segmentation, examining representations, benchmark performance, efficiency, and architectural choices. Across these settings, mask-based representations and the proposed online design deliver strong accuracy with real-time operation.
- Evaluation settings: SiamMask is evaluated on VOT-2016 and VOT-2018 for tracking, and DAVIS-2016, DAVIS-2017, and YouTube-VOS for semi-supervised VOS.The tracking evaluation uses IOU, AP, EAO, accuracy, and robustness; VOS evaluation uses region similarity, contour accuracy, recall, mean, and decay.
- Object representation: mAP@0.5 IOU reaches 85.4 for SiamMask-MBR, improving by +29 and +9.2 points over the two fully-convolutional baselines.At mAP@0.7 IOU, the corresponding improvements widen to +41.6 and +18.4 points.
- Object representation: The MBR strategy for converting masks to rotated boxes offers a significant advantage over axis-aligned bounding boxes.SiamMask achieves the best mIOU across the evaluated box-generation strategies, while SiamMask-Opt has the highest IOU and mAP but requires slow optimization.
- Tracking results: EAO reaches 0.380 at 55 frames per second for the three-branch SiamMask on VOT-2018, significantly outperforming DaSiamRPN.SiamMask-2B reaches 0.334 EAO and is superior to other real-time methods reported in the published literature.
- VOS results: SiamMask is competitive on DAVIS-2016 and DAVIS-2017 while being four times more efficient than the fastest recent VOS methods without fine-tuning.It is almost two orders of magnitude faster than accurate approaches such as OnAVOS or SFL.
- VOS results: SiamMask achieves very low decay for region similarity and contour accuracy, indicating stable performance over time on long sequences.The method operates online without test-sequence adaptation and uses a simple bounding-box initialization.
- Further analysis: Replacing the shared backbone improves performance but reduces speed, while refinement is especially important for contour accuracy FM.SiamMask-2B and SiamMask improve over SiamFC and SiamRPN with the same backbone.
- Further analysis: The two-branch and three-branch variants improve VOT-2018 EAO from 0.251 to 0.265 and from 0.359 to 0.363, respectively.The mask branch is trained but not used during inference in this ablation.
5. Conclusion
SiamMask enables fully-convolutional Siamese trackers to produce class-agnostic binary masks while addressing both visual tracking and semi-supervised video object segmentation. The method uses bounding-box initialization, operates online in real time, and requires no test-sequence adaptation.
- The method addresses both visual object tracking and semi-supervised video object segmentation, achieving better accuracy than state-of-the-art trackers and the fastest speed among VOS methods.
- SiamMask variants use simple bounding-box initialization, operate online, run in real time, and require no adaptation to the test sequence.
A. Architectural details
The architectural details use a ResNet-50 backbone with modified stride and dilation, lightweight network heads, and stacked refinement modules for more accurate masks.
- Both variants use ResNet-50 through the final convolutional layer of the fourth stage as their backbone.
- The backbone reduces output stride to 8 with stride-1 convolutions and increases receptive field using dilated convolutions.
- The network heads use normalization and ReLU in conv5, while conv6 consists only of a 1×1 convolution.
- Mask refinement merges low- and high-resolution features through upsampling layers and skip connections in multiple refinement modules.
- Table 8 provides the backbone architecture and details each building block in square brackets.
B. Further qualitative results
The supplementary figures visualize SiamMask’s refinement and mask-generation process, along with qualitative outputs on tracking and video object segmentation benchmarks.
- Tables 9 and 10 give architectural details for the three-branch and two-branch heads.
- Figure 6 illustrates an example refinement module U3, while Figure 8 schematically shows stacked refinement modules generating masks.
- The mask branch produces score maps at different locations and generates a mask for each RoW.
- During inference, the score branch selects the final mask from the location with the maximum score.
- Figures 9 and 10 provide further qualitative results on VOT-2018 and DAVIS-2016/DAVIS-2017 sequences, respectively.