Joint Depth Prediction and Semantic Segmentation with Multi-View SAM

Abstract
Multi-task approaches to joint depth and segmentation prediction are well-studied for monocular images. Yet, predictions from a single-view are inherently limited, while multiple views are available in many robotics applications. On the other end of the spectrum, video-based and full 3D methods require numerous frames to perform reconstruction and segmentation. With this work we propose a Multi-View Stereo (MVS) technique for depth prediction that benefits from rich semantic features of the Segment Anything Model (SAM). This enhanced depth prediction, in turn, serves as a prompt to our Transformer-based semantic segmentation decoder. We report the mutual benefit that both tasks enjoy in our quantitative and qualitative studies on the ScanNet dataset. Our approach consistently outperforms single-task MVS and segmentation models, along with multi-task monocular methods.
Cite
@inproceedings{joint-depth-prediction-and-semantic-segmentation-with-multi-2024,
author = {Mykhailo Shvets and Dongxu Zhao and Marc Niethammer and Roni Sengupta and Alexander C. Berg},
title = {Joint Depth Prediction and Semantic Segmentation with Multi-View SAM},
booktitle = {WACV},
pages = {1317-1327},
year = {2024},
doi = {10.1109/WACV57701.2024.00136},
url = {https://doi.org/10.1109/WACV57701.2024.00136},
}