NeiEA-NET: Semantic segmentation of large-scale point cloud scene via neighbor enhancement and aggregation

Yongyang Xu, Wei Tang, Ziyin Zeng, Weichao Wu, Jie Wan, Han Guo, Zhong Xie

Abstract: Highlights•A neighbor feature enhancement (NFE) module to adequately and accurately represent local contextual information and enhance feature generalization, which can alleviate the problem of the inaccurate local context information from the KNN-based predefined neighbors.•A neighbor feature aggregation (NFA) module to integrate the most significant local features and the entire local neighbors, which can alleviate the problem that valid information of local neighbors is difficult to be fully captured.•Our method achieves excellent performance on two large-scale benchmarks, including S3DIS and Toronto3D.