Orientation-Encoding CNN for Point Cloud Classification and Segmentation

• Main Authors: Hongbin Lin, Wu Zheng, Xiuping Peng
• Format: Article
• Language: English
• Published: MDPI AG 2021-08-01
• Series: Machine Learning and Knowledge Extraction
• Subjects: point clouds; orientation-encoding (OE) convolution; local geometric feature; classification; segmentation
• Online Access: https://www.mdpi.com/2504-4990/3/3/31

With the introduction of effective and general deep learning network frameworks, deep learning based methods have achieved remarkable success in various visual tasks. However, there are still tough challenges in applying them to convolutional neural networks due to the lack of a potential rule structure of point clouds. Therefore, by taking the original point clouds as the input data, this paper proposes an orientation-encoding (OE) convolutional module and designs a convolutional neural network for effectively extracting local geometric features of point sets. By searching for the same number of points in 8 directions and arranging them in order in 8 directions, the OE convolution is then carried out according to the number of points in the direction, which realizes the effective feature learning of the local structure of the point sets. Further experiments on diverse datasets show that the proposed method has competitive performance on classification and segmentation tasks of point sets.