Query refinement transformer for 3d instance segmentation

J Lu, J Deng, C Wang, J He… - Proceedings of the IEEE …, 2023 - openaccess.thecvf.com
J Lu, J Deng, C Wang, J He, T Zhang
Proceedings of the IEEE/CVF International Conference on …, 2023openaccess.thecvf.com
Abstract 3D instance segmentation aims to predict a set of object instances in a scene and
represent them as binary foreground masks with corresponding semantic labels. However,
object instances are diverse in shape and category, and point clouds are usually sparse,
unordered, and irregular, which leads to a query sampling dilemma. Besides, noise
background queries interfere with proper scene perception and accurate instance
segmentation. To address the above issues, we propose a Query Refinement Transformer …
Abstract
3D instance segmentation aims to predict a set of object instances in a scene and represent them as binary foreground masks with corresponding semantic labels. However, object instances are diverse in shape and category, and point clouds are usually sparse, unordered, and irregular, which leads to a query sampling dilemma. Besides, noise background queries interfere with proper scene perception and accurate instance segmentation. To address the above issues, we propose a Query Refinement Transformer termed QueryFormer. The key to our approach is to exploit a query initialization module to optimize the initialization process for the query distribution with a high coverage and low repetition rate. Additionally, we design an affiliated transformer decoder that suppresses the interference of noise background queries and helps the foreground queries focus on instance discriminative parts to predict final segmentation results. Extensive experiments on ScanNetV2 and S3DIS datasets show that our QueryFormer can surpass state-of-the-art 3D instance segmentation methods.
openaccess.thecvf.com
以上显示的是最相近的搜索结果。 查看全部搜索结果