SemanticFusion: Dense 3D Semantic Mapping with Convolutional Neural Networks
About
Ever more robust, accurate and detailed mapping using visual sensing has proven to be an enabling factor for mobile robots across a wide variety of applications. For the next level of robot intelligence and intuitive user interaction, maps need extend beyond geometry and appearence - they need to contain semantics. We address this challenge by combining Convolutional Neural Networks (CNNs) and a state of the art dense Simultaneous Localisation and Mapping (SLAM) system, ElasticFusion, which provides long-term dense correspondence between frames of indoor RGB-D video even during loopy scanning trajectories. These correspondences allow the CNN's semantic predictions from multiple view points to be probabilistically fused into a map. This not only produces a useful semantic 3D map, but we also show on the NYUv2 dataset that fusing multiple predictions leads to an improvement even in the 2D semantic labelling over baseline single frame predictions. We also show that for a smaller reconstruction dataset with larger variation in prediction viewpoint, the improvement over single frame segmentation increases. Our system is efficient enough to allow real-time interactive use at frame-rates of approximately 25Hz.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | NYUD v2 13-class (test) | Mean Accuracy63.6 | 14 | |
| Semantic segmentation | NYUv2 13-class labeling | Accuracy59.2 | 12 | |
| 2D Semantic Segmentation | NYU2 11-class task | Mean Accuracy63.6 | 7 | |
| 3D Semantic Segmentation | ScanNet open v2 (test) | mAP51.8 | 4 |