ET-PointPillars: improved PointPillars for 3D object detection based on optimized voxel downsampling
Yiyi Liu,Zhengyi Yang,JianLin Tong,Jiajia Yang,Jiongcheng Peng,Lihang Zhang,Wangxin Cheng
DOI: https://doi.org/10.1007/s00138-024-01538-y
IF: 2.983
2024-04-22
Machine Vision and Applications
Abstract:The preprocessing of point cloud data has always been an important problem in 3D object detection. Due to the large volume of point cloud data, voxelization methods are often used to represent the point cloud while reducing data density. However, common voxelization randomly selects sampling points from voxels, which often fails to represent local spatial features well due to noise. To preserve local features, this paper proposes an optimized voxel downsampling(OVD) method based on evidence theory. This method uses fuzzy sets to model basic probability assignments (BPAs) for each candidate point, incorporating point location information. It then employs evidence theory to fuse the BPAs and determine the selected sampling points. In the PointPillars 3D object detection algorithm, the point cloud is partitioned into pillars and encoded using each pillar's points. Convolutional neural networks are used for feature extraction and detection. Another contribution is the proposed improved PointPillars based on evidence theory (ET-PointPillars) by introducing an OVD-based feature point sampling module in the PointPillars' pillar feature network, which can select feature points in pillars using the optimized method, computes offsets to these points, and adds them as features to facilitate learning more object characteristics, improving traditional PointPillars. Experiments on the KITTI datasets validate the method's ability to preserve local spatial features. Results showed improved detection precision, with a average increase for pedestrians and cyclists on KITTI.
computer science, cybernetics, artificial intelligence,engineering, electrical & electronic