STD2P: RGBD Semantic Segmentation Using Spatio-Temporal Data-Driven Pooling
About
We propose a novel superpixel-based multi-view convolutional neural network for semantic image segmentation. The proposed network produces a high quality segmentation of a single image by leveraging information from additional views of the same scene. Particularly in indoor videos such as captured by robotic platforms or handheld and bodyworn RGBD cameras, nearby video frames provide diverse viewpoints and additional context of objects and scenes. To leverage such information, we first compute region correspondences by optical flow and image boundary-based superpixels. Given these region correspondences, we propose a novel spatio-temporal pooling layer to aggregate information over space and time. We evaluate our approach on the NYU--Depth--V2 and the SUN3D datasets and compare it to various state-of-the-art single-view and multi-view approaches. Besides a general improvement over the state-of-the-art, we also show the benefits of making use of unlabeled frames during training for multi-view as well as single-view prediction.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | NYU v2 (test) | mIoU40.1 | 248 | |
| Semantic segmentation | NYU Depth V2 (test) | mIoU40.1 | 172 | |
| Semantic segmentation | NYUDv2 40-class (test) | mIoU40.1 | 99 | |
| Semantic segmentation | NYUD v2 13-class (test) | Mean Accuracy68.4 | 14 | |
| Semantic segmentation | SUN3D 33-class (test) | Pixel Accuracy65.5 | 10 | |
| Video Semantic Segmentation | NYUDepth | mIoU40.1 | 8 | |
| Semantic segmentation | NYUDv2 4-class (test) | Mean Accuracy82.5 | 7 |