# Enhancing Point Features with Spatial Information for Point-Based 3D Object Detection

Authors: Huaijin Liu; Jixiang Du; Yong Zhang; Hongbo Zhang
Journal: Scientific Programming
 (2021)

---

## Abstract

Currently, there are many kinds of voxel-based multisensor 3D object detectors, while point-based multisensor 3D object detectors have not been fully studied. In this paper, we propose a new 3D two-stage object detection method based on point cloud and image fusion to improve the detection accuracy. To address the problem of insufficient semantic information of point cloud, we perform multiscale deep fusion of LiDAR point and camera image in a point-wise manner to enhance point features. Due to the imbalance of LiDAR points, the object point cloud in the long-distance area is sparse. We design a point cloud completion module to predict the spatial shape of objects in the candidate boxes and extract the structural information to improve the feature representation ability to further refine the boxes. The framework is evaluated on widely used KITTI and SUN-RGBD dataset. Experimental results show that our method outperforms all state-of-the-art point-based 3D object detection methods and has comparable performance to voxel-based methods as well.


---

## Body

## 1. Introduction

3D object detection is particularly useful in autonomous driving applications, because various types of dynamic objects must be recognized in the driving environment, such as surrounding vehicles, pedestrians, and cyclists. In recent years, various 3D detectors using LiDAR point clouds have been proposed, including PointRCNN [1], Part- A2 [2], PV-RCNN++ [3], 3DSSD [4], and CIA-SSD [5]. Although LiDAR points can capture the three-dimensional structure of an object and contain accurate depth information, they do not have sufficient semantic information and have the problem of point sparsity. Compared with LiDAR point clouds, RGB images have more regular and dense data format and have richer semantic information to distinguish between vehicles and backgrounds. Therefore, some research works [6, 7] try to estimate the position and size of objects through monocular or stereo images. However, the biggest challenge of 3D object detection based on camera image is that it cannot get accurate depth information, which is very important for 3D object detection. Considering that the representation under different sensor views have their own shortcomings, and for the 3D object detector of automatic driving, only one view input is not enough. This prompts us to design an effective framework to integrate features from different perspectives to achieve accurate 3D object detection. Early multisensor feature fusion methods take RGB image, front view, and bird’s eye view (BEV) as input and then directly combine and merge the features by cropping and resizing to generate 3D candidate boxes, such as MVF [8] and AVOD [9], but they ignore the different perspectives of image and BEV. In order to reduce the accuracy loss caused by different viewing angles, ContFuse [10] uses continuous convolution to improve feature fusion, and MVAF-Net [11] uses bilinear interpolation to correct features. Although continuous convolution or bilinear interpolation is used to modify alignment to overcome the challenges of different perspectives, quantifying point cloud 3D structures into BEV pseudoimages to fusion image features will inevitably suffer a loss of accuracy. There are also some research works [12, 13] using 3D frustum projected by 2D bounding boxes to estimate 3D bounding boxes, but these methods require additional 2D annotations and their performance is limited by 2D detectors. The above multisensor feature fusion methods all transform point clouds from sparse formation to compact representation by projecting them into images or subdividing them into uniformly distributed voxel. We call these methods voxel-based multimodal feature fusion methods, which voxelize the entire point cloud. However, the voxel-based feature fusion method will inevitably lose some information and is relatively sensitive to voxel parameters. There are also some methods that directly perform image feature fusion on LiDAR point cloud, instead of performing image feature fusion with BEV of the point cloud or voxelized pseudoimage of the point cloud. These methods are called point-based multimodal feature fusion methods. For example, PI-RCNN [14] directly fuses image features and point features, and EPNet [15] and MOT [16] perform deep fusion between point features and image features. In addition, since object detection serves the perception system of autonomous vehicle, the farther the object detected is, the more the time left for the decision planning system is, and the safer the autonomous vehicle will be. However, due to the imbalance of point clouds, the point clouds of the short-distance object are denser, and the point clouds of the long-distance object are sparse, which contains less spatial information, thus increasing the difficulty of detecting the distant object. In order to improve the detection accuracy of difficult cases, SIENet [17] predicts the shape of distant objects through point completion network to enhance the spatial structure information. Inspired by some multitask work (EPNet and SIENet), this paper proposes a point-based multimodal fusion 3D object detection method with enhanced spatial structure.The main contributions of this paper are as follows: (1) we design a new backbone network for multimodal feature fusion, which combines LiDAR points and camera images in a point-wise manner to enhance point features without point cloud voxelization and image annotation. (2) A spatial structure enhancement module is proposed to predict the shape of object in the candidate box and learn structural information to further refine box. (3) We propose a new two-stage 3D object detection framework based on point cloud and image fusion. The test results on the KITTI benchmark show that the accuracy of our method is higher than all the current multisensor-based 3D object detection methods.


## 2. Related Work

3D object detection based on LiDAR: due to the sparsity and irregularity of LIDAR point cloud, traditional convolutional neural networks (CNN) cannot be directly applied to LIDAR point cloud. Many algorithms have tried various point cloud representation methods to solve this problem. Currently, there are three types of point cloud representation for the input of the 3D detector. (1) Based on the voxel representation, this method converts point clouds into regular grids through voxel transformation, so that 3D CNN can directly apply this representation. SECOND [18] divides the point cloud into voxel representations and uses sparse convolution to learn voxel features to generate 3D bounding boxes. PointPillars [19] converts point clouds into pseudoimages, eliminating the time-consuming 3D convolution operations. Fast-PointRCNN [20] introduces the attention mechanism to enhance the positioning ability of the network. The ROI-aware pooling proposed by Part- A2 [2] refines the candidate box and improves the 3D detection accuracy. The voxel-based method has high perceptual ability, but it will cause information loss during the voxelization process of point cloud. And the storage and computing efficiency of 3D CNN are very low. (2) Based on the point representation, this method does not need to transform the original point cloud and directly uses PointNet++ [21] to process the original point cloud to obtain global features, thus retaining the original geometric information as much as possible. F-PointNet [12] proposes the application of PointNet++ [21] to 3D detection based on the cropped point cloud of 2D image bounding box. Point-RCNN [1] is the first point-based 3D object detection method that only uses point cloud as network input. 3DSSD [4] proposed a lightweight and efficient point-based single-stage 3D object detection framework, which has a good balance between accuracy and speed. (3) Point-voxel joint representation method takes points and voxels as inputs and fuses the features of points and voxels at different stages of the network for 3D object detection, such as Part- A2 [2] and PV-RCNN++ [3]. These methods can use voxel-based perception capabilities (i.e., 3D sparse convolution) and point-based geometric structure capabilities (i.e., set abstraction) to achieve high computational efficiency and flexible receiving field, thereby improving 3D detection performance.3D object detection based on multiple sensors: in recent years, great progress has been made in the research of multisensors such as camera image and LiDAR. AVOD [9] uses RGB image and BEV as input, proposes a feature pyramid skeleton to extract features in BEV, and combines features from BEV feature map and RGB feature map through cropping and resizing operations. ContFuse [10] applies continuous convolution to overcome the problem of different viewing angles between image and BEV. MVAF-Net [11] proposes a multiview adaptive fusion module to enhance feature fusion among image, front view, and BEV. The above methods all try to fuse the features of image and BEV, but quantifying the point cloud 3D structure into BEV pseudoimage to fuse image features will inevitably suffer accuracy loss. F-PointNet [12] uses 3D frustum projected from 2D bounding boxes to estimate 3D bounding boxes, but this method requires additional 2D annotations, and their performance is limited by 2D detectors. There are also some methods that directly perform image feature fusion on the LiDAR point cloud rather than the LiDAR BEV or the voxelized pseudoimage of the point cloud. PI-RCNN [14] directly attaches the image semantic segmentation information to the LiDAR point cloud through the transformation matrix and then uses the LiDAR detector for 3D object detection. EPNet [15] and MOT [16] establish a deep fusion between the point cloud feature extractor and the image feature extractor to enhance the point cloud features. Although various sensor fusion networks have been proposed, they are not easily superior to LiDAR detectors because the fusion of multiview features will bring interference and noise.


## 3. Our Framework

In this section, we introduce a new two-stage 3D object detection framework based on point cloud and image fusion. Firstly, we describe our proposed multiscale deep fusion strategy and proposal generation layers. Next, we propose a spatial structure prediction network, including point cloud region pooling, spatial structure enhancement, and refined regression head. Finally, the loss function is discussed. Our overall framework is shown in Figure1.Figure 1
Description of the 3D object detection framework based on point cloud and image fusion. The whole framework consists of two stages. Stage 1 uses two-stream deep fusion RPN to extract backbone features and generate proposal boxes. Stage 2 generates high-quality 3D boxes through the spatial structure enhancement module.


### 3.1. Multiscale Feature Fusion RPN

As shown in Figure2, our multiscale feature fusion RPN consists of a point branch and an image branch. Specifically, we first use a four-layer four-scale PointNet++ to extract point features from the point cloud. Meanwhile, the image branch extracts semantic features from the image through a four-layer four-scale Unet [22] segmentation network. Finally, the proposed adaptive attention fusion (AAF) module is used to fuse the point features at different scales with corresponding image semantic features to enhance the point features.Figure 2
Description of multiscale feature fusion RPN composed of a point branch and an image branch. We use multiple AAF modules to enhance LiDAR point features with corresponding image semantic features at multiple scales.


#### 3.1.1. Point Branch

The point branch takes LiDAR point cloud as input and generates 3D candidate boxes. The point branch is composed of four paired set abstraction (SA) and feature propagation (FP) layers for extracting point cloud features. SA consists of farthest point sampling (FPS) layer, multiscale grouping (MSG) layer, and PointNet layer, which are used for downsampling points to improve efficiency and expand the receptive field. FP consists of bilinear interpolation and multilayer perception (MLP), which is applied to broadcast feature for dropped points during the downsampling process to recover all points. Due to insufficient semantic information of LiDAR point cloud, we use LI-Fusion module [18] to fuse rich image semantic features and point features. In addition, multiscale deep fusion of point clouds and images can further enrich the point semantic features and obtain compact and discriminative feature representations. The multiscale feature fusion method is shown in Figure 2.


#### 3.1.2. Image Branch

In order to perform multiscale semantic feature fusion, we choose the lightweight semantic segmentation network Unet that also has an encoder and decoder for image semantic feature extraction. Unet consists of four convolution blocks and four upsampling layers. Each convolution block has two repeated3×3 convolution layers and a 2×2 maximum pooling layer. In order to obtain strong semantic features and balance GPU memory, we fine-tuned the convolution block of Unet. Our convolution block consists of two repeated 3×3 convolution layers (stride 1, padding 1) and one 3×3 convolution layer (stride 2, padding 1). Each of the first two convolution layers is followed by a batch normalization layer and a ReLU activation function, as shown in Figure 3.Figure 3
Details of the image backbone network. Conv2d (cin, cout,k, s, p) represents 2D convolution, and DeConv2d (cin, cout, k, s) represents 2D deconvolution, where cin, cout, k, s, and p represent the number of input channels, the number of output channels, kernel size, stride, and padding, respectively. Each convolution block consists of Convolution, BatchNorm, and ReLU.


#### 3.1.3. Adaptive Attention Fusion Module

In order to fuse data from two different views, we first use the projection method to establish the relationship between LiDAR points and image pixels. Then, we obtain the semantic features of each point through grid sampling. Finally, the proposed adaptive attention fusion (AAF) module is used to perform feature fusion. Specifically, we take each point coordinatepx,y,z through the projection matrix M to generate the corresponding image coordinate p′x′,y′, which can be written as(1)p′=M⋅p,where M is the internal parameters of the camera and the size is 3×4. Note that we convert p and p′ into four-dimensional and three-dimensional vectors in homogeneous coordinates in projection formula (1). After establishing the corresponding relationship, we use the grid sample function of Pytorch framework to obtain the semantic features of each point on the image. Because the projection point may fall between adjacent pixels, the bilinear interpolation method needs to be used to obtain the image feature at the continuous coordinates, which can be written as(2)Fp=ℬFNp′,where Fp is the corresponding image feature for point p, ℬ is the bilinear interpolation function, and FNp′ is the image feature of the adjacent pixels of the projection point p′. Finally, in order to better integrate point cloud features and image features, we design an adaptive attention fusion module to suppress the interference of noninterested areas and extract effective information for fusion, as shown in Figure 4. The adaptive attention fusion module can be expressed as follows:(3)FE=τFC1FP⊕FC2FI,FPA=FP⊗σFC3FE,FIA=FI⊗σFC4FE,FPI=ConcatFPA,FIA,where FP and FI represent point cloud features and point-wise image semantic features, FE represents extended features, FPA and FIA represent two-branch attention features, FPI represents fusion features, FC represents fully connected layer, ⊕ represents element-wise addition, ⊗ represents element-wise multiplication, σ represents the Sigmoid activation function, τ represents the Tanh activation function, and Concat represents the concatenation operation.Figure 4
Illustration of the AAF module. FC represents fully connected layer,⊕ represents element-wise addition, and ⊗ represents element-wise multiplication.


### 3.2. Spatial Structure Enhancement Module

For each candidate box generated in RPN stage, the denser the foreground point set is, the more spatial the information retained is. Therefore, the central idea of our spatial information enhancement module is to predict the complete shapes of candidate objects and extract structural information to enhance feature representation. To this end, we need to solve two subtasks, namely, how to predict the spatial shape, and how to extract the spatial structural information and integrate it into the model to further refine the candidate box.


#### 3.2.1. Point Cloud Region Pooling

After obtaining 3D bounding box proposals, we use RoI Pooling [1] to optimize the box locations and orientations. Specifically, 512 candidate regions of RPN are sampled through NMS to obtain 64 candidate regions of RCNN. For each 3D box bi=xi,yi,zi,hi,wi,li,θi, we slightly enlarge it to create a new 3D box bi′=xi,yi,zi,hi+μ,wi+μ,li+μ,θi, so as to obtain additional context information, where xi,yi,zi is the center location of object bi, hi,wi,li is the size of object bi, θi is the object orientation of the bird’s view, and μ is a constant used to expand the box size. For each point, through the segmentation mask we perform an internal/external testing, to determine whether the point is within the expanded bounding box proposal bi′. If it is an internal point, the point and its features would be retained to refine the box bi. Finally, we will get 512 points for each candidate box and encode them to get the pooling feature Fp∈ℜ512×C1, where C1 represents the number of channels.


#### 3.2.2. Spatial Shape Prediction

The foreground points of the candidate box constitute a shape describing semantic clues; however, this shape is usually incomplete. Therefore, based on the point completion framework PCN [23], we design a spatial structure prediction network to complete the missing part of the object in the candidate box. As shown in Figure 5, the network takes incomplete points as input and predicts the corresponding dense shape through the encoder-decoder. The encoder consists of two simple PointNet units (SharedMLP + Maxpool), each SharedMLP consisting of a 1×1 convolution layer, a BN layer, a ReLU layer, and a 1×1 convolution layer. The number of convolution output channels for the first SharedMLP is (128, 256), and the number of convolution output channels for the second SharedMLP is (512, 1024). The decoder consists of two stacked fully connected layers (Linear + BN + ReLU) and one fully connected layer (Linear), and the output is a 1024×3 matrix. The number of output channels for the three fully connected layers is (1024, 1024, 3∗1023). Unlike the coarse-to-fine pipeline in PCN, we believe that the coarse output is effective for subsequent processing, so we remove the fine output branch, thus saving GPU memory. In order to reduce the burden of training, we download the KITTI [23] car data set and trained our spatial shape prediction network in advance.Figure 5
Graphical spatial structure prediction network. The network takes the incomplete points as input and predicts the corresponding complete shape.Figure6 shows part of the visualization results of our spatial structure prediction model. It can be seen from the figure that our spatial structure prediction model performs well on automobiles and has a good generalization prospect.Figure 6
Visualization of spatial shape prediction of candidate objects. The original LiDAR point cloud (top) and the corresponding prediction results (bottom) for each object.


#### 3.2.3. Structure Information Extraction and Fusion

To obtain the local and global context from the predicted spatial shapes, we use a PointNet++ [21] module to extract the structural information. First, we use the FPS algorithm to select 512 points from the predicted shape. Then for each point, we use the Ball Query algorithm to generate a local area. Finally, the PointNet units are applied to capture the local area feature C2 of each point, thereby obtaining the enhanced features Fs∈ℜ512×C2. In the refinement subnetwork part, we use a similar 3D box refinement network of PointRCNN [1] to further refine the box and confidence. The input of refining subnetwork consists of the canonical transformation coordinates of each pooling point, the pooling features, and the extracted spatial structure features. Since the pooling features and the spatial structure features come from different patterns, connecting them without any additional processing may cause interference. In order to better fusion spatial structure features and pooling features, we adopt the perspective-channel attention fusion [24] to obtain merged feature Fm∈ℜ512×C1+C2.


### 3.3. Loss Function

The proposed network is trained in an end-to-end manner. Our overall lossesLtotal include the two-stream RPN loss Lrpn in stage 1 and the box refining network loss Lrcnn in stage 2 as follows:(4)Ltotal=ωrpnLrpn+ωrcnnLrcnn,where ωrpn and ωrcnn are the coefficients that control the balance weight; we set the parameters ωrpn = 1.0 and ωrcnn = 1.0. Lrpn and Lrcnn adopt similar optimization objectives, including classification loss, regression loss, and consistency enhancement loss. For classification loss at the RPN stage, we use focal loss similar to [25] to balance positive and negative samples:(5)Lclsrpn=−1N∑i=1Nα1−y^iγlogy^i,ifyi=1,αyi+1−α1−yiy^iγlog1−y^i,otherwise,where yi is the target classification label, y^i is the positive sample prediction probability, N represents the number of targets, and α and γ are focal loss hyperparameters. For the regression loss in the RPN stage, we adopt a bin-based regression loss similar to [1] to regress the center point x,y,z, size l,h,w, and orientation θ:(6)Lresrpn=−1N∑i=1N∑u∈x,z,θFclsbin^u,binu+∑u∈x,y,z,h,w,l,θFregres^u,resu,where Fcls denotes the cross-entropy classification loss, Freg denotes the smooth 1 loss, binu and resu denote the bins and residuals of the ground truth, and bin^u and res^u denote the predicted bins and residuals of the ground truth. In addition, in order to improve the consistency of localization confidence and classification confidence, we add a consistency enhancement loss:(7)Lcerpn=−1N∑i=1Nlogci×Bi∩BigtBi∪Bigt,where Bi represents the predicted bounding box, Bigt represents the ground truth, and ci represents the classification confidence of the predicted box. In summary, Lrpn is a weighted sum of the three loss functions:(8)Lrpn=ωclsLclsrpn+ωregLregrpn+ωceLcerpn,where ωcls, ωreg, and ωce are used to control the balance coefficient of the importance degree of loss. We set the parameters ωcls = 1.0, ωreg = 1.0, and ωce = 5.0. Similarly, RCNN loss also includes classification loss, regression loss, and consistency enhancement loss. For RCNN classification loss, we adopt binary cross entropy loss:(9)Lclsrcnn=−1N∑i=1Nyilogy^i+1−yilog1−y^i,where yi is the target classification label, y^i is the target prediction probability, and N is the number of targets. RCNN regression loss Lregrcnn and consistency enhancement loss Lcercnn are defined in the same way as RPN. The weighted sum of the three loss functions of rcnn:(10)Lrcnn=ωclsLclsrcnn+ωregLregrcnn+ωceLcercnn.


## 3.1. Multiscale Feature Fusion RPN

As shown in Figure2, our multiscale feature fusion RPN consists of a point branch and an image branch. Specifically, we first use a four-layer four-scale PointNet++ to extract point features from the point cloud. Meanwhile, the image branch extracts semantic features from the image through a four-layer four-scale Unet [22] segmentation network. Finally, the proposed adaptive attention fusion (AAF) module is used to fuse the point features at different scales with corresponding image semantic features to enhance the point features.Figure 2
Description of multiscale feature fusion RPN composed of a point branch and an image branch. We use multiple AAF modules to enhance LiDAR point features with corresponding image semantic features at multiple scales.


### 3.1.1. Point Branch

The point branch takes LiDAR point cloud as input and generates 3D candidate boxes. The point branch is composed of four paired set abstraction (SA) and feature propagation (FP) layers for extracting point cloud features. SA consists of farthest point sampling (FPS) layer, multiscale grouping (MSG) layer, and PointNet layer, which are used for downsampling points to improve efficiency and expand the receptive field. FP consists of bilinear interpolation and multilayer perception (MLP), which is applied to broadcast feature for dropped points during the downsampling process to recover all points. Due to insufficient semantic information of LiDAR point cloud, we use LI-Fusion module [18] to fuse rich image semantic features and point features. In addition, multiscale deep fusion of point clouds and images can further enrich the point semantic features and obtain compact and discriminative feature representations. The multiscale feature fusion method is shown in Figure 2.


### 3.1.2. Image Branch

In order to perform multiscale semantic feature fusion, we choose the lightweight semantic segmentation network Unet that also has an encoder and decoder for image semantic feature extraction. Unet consists of four convolution blocks and four upsampling layers. Each convolution block has two repeated3×3 convolution layers and a 2×2 maximum pooling layer. In order to obtain strong semantic features and balance GPU memory, we fine-tuned the convolution block of Unet. Our convolution block consists of two repeated 3×3 convolution layers (stride 1, padding 1) and one 3×3 convolution layer (stride 2, padding 1). Each of the first two convolution layers is followed by a batch normalization layer and a ReLU activation function, as shown in Figure 3.Figure 3
Details of the image backbone network. Conv2d (cin, cout,k, s, p) represents 2D convolution, and DeConv2d (cin, cout, k, s) represents 2D deconvolution, where cin, cout, k, s, and p represent the number of input channels, the number of output channels, kernel size, stride, and padding, respectively. Each convolution block consists of Convolution, BatchNorm, and ReLU.


### 3.1.3. Adaptive Attention Fusion Module

In order to fuse data from two different views, we first use the projection method to establish the relationship between LiDAR points and image pixels. Then, we obtain the semantic features of each point through grid sampling. Finally, the proposed adaptive attention fusion (AAF) module is used to perform feature fusion. Specifically, we take each point coordinatepx,y,z through the projection matrix M to generate the corresponding image coordinate p′x′,y′, which can be written as(1)p′=M⋅p,where M is the internal parameters of the camera and the size is 3×4. Note that we convert p and p′ into four-dimensional and three-dimensional vectors in homogeneous coordinates in projection formula (1). After establishing the corresponding relationship, we use the grid sample function of Pytorch framework to obtain the semantic features of each point on the image. Because the projection point may fall between adjacent pixels, the bilinear interpolation method needs to be used to obtain the image feature at the continuous coordinates, which can be written as(2)Fp=ℬFNp′,where Fp is the corresponding image feature for point p, ℬ is the bilinear interpolation function, and FNp′ is the image feature of the adjacent pixels of the projection point p′. Finally, in order to better integrate point cloud features and image features, we design an adaptive attention fusion module to suppress the interference of noninterested areas and extract effective information for fusion, as shown in Figure 4. The adaptive attention fusion module can be expressed as follows:(3)FE=τFC1FP⊕FC2FI,FPA=FP⊗σFC3FE,FIA=FI⊗σFC4FE,FPI=ConcatFPA,FIA,where FP and FI represent point cloud features and point-wise image semantic features, FE represents extended features, FPA and FIA represent two-branch attention features, FPI represents fusion features, FC represents fully connected layer, ⊕ represents element-wise addition, ⊗ represents element-wise multiplication, σ represents the Sigmoid activation function, τ represents the Tanh activation function, and Concat represents the concatenation operation.Figure 4
Illustration of the AAF module. FC represents fully connected layer,⊕ represents element-wise addition, and ⊗ represents element-wise multiplication.


## 3.1.1. Point Branch

The point branch takes LiDAR point cloud as input and generates 3D candidate boxes. The point branch is composed of four paired set abstraction (SA) and feature propagation (FP) layers for extracting point cloud features. SA consists of farthest point sampling (FPS) layer, multiscale grouping (MSG) layer, and PointNet layer, which are used for downsampling points to improve efficiency and expand the receptive field. FP consists of bilinear interpolation and multilayer perception (MLP), which is applied to broadcast feature for dropped points during the downsampling process to recover all points. Due to insufficient semantic information of LiDAR point cloud, we use LI-Fusion module [18] to fuse rich image semantic features and point features. In addition, multiscale deep fusion of point clouds and images can further enrich the point semantic features and obtain compact and discriminative feature representations. The multiscale feature fusion method is shown in Figure 2.


## 3.1.2. Image Branch

In order to perform multiscale semantic feature fusion, we choose the lightweight semantic segmentation network Unet that also has an encoder and decoder for image semantic feature extraction. Unet consists of four convolution blocks and four upsampling layers. Each convolution block has two repeated3×3 convolution layers and a 2×2 maximum pooling layer. In order to obtain strong semantic features and balance GPU memory, we fine-tuned the convolution block of Unet. Our convolution block consists of two repeated 3×3 convolution layers (stride 1, padding 1) and one 3×3 convolution layer (stride 2, padding 1). Each of the first two convolution layers is followed by a batch normalization layer and a ReLU activation function, as shown in Figure 3.Figure 3
Details of the image backbone network. Conv2d (cin, cout,k, s, p) represents 2D convolution, and DeConv2d (cin, cout, k, s) represents 2D deconvolution, where cin, cout, k, s, and p represent the number of input channels, the number of output channels, kernel size, stride, and padding, respectively. Each convolution block consists of Convolution, BatchNorm, and ReLU.


## 3.1.3. Adaptive Attention Fusion Module

In order to fuse data from two different views, we first use the projection method to establish the relationship between LiDAR points and image pixels. Then, we obtain the semantic features of each point through grid sampling. Finally, the proposed adaptive attention fusion (AAF) module is used to perform feature fusion. Specifically, we take each point coordinatepx,y,z through the projection matrix M to generate the corresponding image coordinate p′x′,y′, which can be written as(1)p′=M⋅p,where M is the internal parameters of the camera and the size is 3×4. Note that we convert p and p′ into four-dimensional and three-dimensional vectors in homogeneous coordinates in projection formula (1). After establishing the corresponding relationship, we use the grid sample function of Pytorch framework to obtain the semantic features of each point on the image. Because the projection point may fall between adjacent pixels, the bilinear interpolation method needs to be used to obtain the image feature at the continuous coordinates, which can be written as(2)Fp=ℬFNp′,where Fp is the corresponding image feature for point p, ℬ is the bilinear interpolation function, and FNp′ is the image feature of the adjacent pixels of the projection point p′. Finally, in order to better integrate point cloud features and image features, we design an adaptive attention fusion module to suppress the interference of noninterested areas and extract effective information for fusion, as shown in Figure 4. The adaptive attention fusion module can be expressed as follows:(3)FE=τFC1FP⊕FC2FI,FPA=FP⊗σFC3FE,FIA=FI⊗σFC4FE,FPI=ConcatFPA,FIA,where FP and FI represent point cloud features and point-wise image semantic features, FE represents extended features, FPA and FIA represent two-branch attention features, FPI represents fusion features, FC represents fully connected layer, ⊕ represents element-wise addition, ⊗ represents element-wise multiplication, σ represents the Sigmoid activation function, τ represents the Tanh activation function, and Concat represents the concatenation operation.Figure 4
Illustration of the AAF module. FC represents fully connected layer,⊕ represents element-wise addition, and ⊗ represents element-wise multiplication.


## 3.2. Spatial Structure Enhancement Module

For each candidate box generated in RPN stage, the denser the foreground point set is, the more spatial the information retained is. Therefore, the central idea of our spatial information enhancement module is to predict the complete shapes of candidate objects and extract structural information to enhance feature representation. To this end, we need to solve two subtasks, namely, how to predict the spatial shape, and how to extract the spatial structural information and integrate it into the model to further refine the candidate box.


### 3.2.1. Point Cloud Region Pooling

After obtaining 3D bounding box proposals, we use RoI Pooling [1] to optimize the box locations and orientations. Specifically, 512 candidate regions of RPN are sampled through NMS to obtain 64 candidate regions of RCNN. For each 3D box bi=xi,yi,zi,hi,wi,li,θi, we slightly enlarge it to create a new 3D box bi′=xi,yi,zi,hi+μ,wi+μ,li+μ,θi, so as to obtain additional context information, where xi,yi,zi is the center location of object bi, hi,wi,li is the size of object bi, θi is the object orientation of the bird’s view, and μ is a constant used to expand the box size. For each point, through the segmentation mask we perform an internal/external testing, to determine whether the point is within the expanded bounding box proposal bi′. If it is an internal point, the point and its features would be retained to refine the box bi. Finally, we will get 512 points for each candidate box and encode them to get the pooling feature Fp∈ℜ512×C1, where C1 represents the number of channels.


### 3.2.2. Spatial Shape Prediction

The foreground points of the candidate box constitute a shape describing semantic clues; however, this shape is usually incomplete. Therefore, based on the point completion framework PCN [23], we design a spatial structure prediction network to complete the missing part of the object in the candidate box. As shown in Figure 5, the network takes incomplete points as input and predicts the corresponding dense shape through the encoder-decoder. The encoder consists of two simple PointNet units (SharedMLP + Maxpool), each SharedMLP consisting of a 1×1 convolution layer, a BN layer, a ReLU layer, and a 1×1 convolution layer. The number of convolution output channels for the first SharedMLP is (128, 256), and the number of convolution output channels for the second SharedMLP is (512, 1024). The decoder consists of two stacked fully connected layers (Linear + BN + ReLU) and one fully connected layer (Linear), and the output is a 1024×3 matrix. The number of output channels for the three fully connected layers is (1024, 1024, 3∗1023). Unlike the coarse-to-fine pipeline in PCN, we believe that the coarse output is effective for subsequent processing, so we remove the fine output branch, thus saving GPU memory. In order to reduce the burden of training, we download the KITTI [23] car data set and trained our spatial shape prediction network in advance.Figure 5
Graphical spatial structure prediction network. The network takes the incomplete points as input and predicts the corresponding complete shape.Figure6 shows part of the visualization results of our spatial structure prediction model. It can be seen from the figure that our spatial structure prediction model performs well on automobiles and has a good generalization prospect.Figure 6
Visualization of spatial shape prediction of candidate objects. The original LiDAR point cloud (top) and the corresponding prediction results (bottom) for each object.


### 3.2.3. Structure Information Extraction and Fusion

To obtain the local and global context from the predicted spatial shapes, we use a PointNet++ [21] module to extract the structural information. First, we use the FPS algorithm to select 512 points from the predicted shape. Then for each point, we use the Ball Query algorithm to generate a local area. Finally, the PointNet units are applied to capture the local area feature C2 of each point, thereby obtaining the enhanced features Fs∈ℜ512×C2. In the refinement subnetwork part, we use a similar 3D box refinement network of PointRCNN [1] to further refine the box and confidence. The input of refining subnetwork consists of the canonical transformation coordinates of each pooling point, the pooling features, and the extracted spatial structure features. Since the pooling features and the spatial structure features come from different patterns, connecting them without any additional processing may cause interference. In order to better fusion spatial structure features and pooling features, we adopt the perspective-channel attention fusion [24] to obtain merged feature Fm∈ℜ512×C1+C2.


## 3.2.1. Point Cloud Region Pooling

After obtaining 3D bounding box proposals, we use RoI Pooling [1] to optimize the box locations and orientations. Specifically, 512 candidate regions of RPN are sampled through NMS to obtain 64 candidate regions of RCNN. For each 3D box bi=xi,yi,zi,hi,wi,li,θi, we slightly enlarge it to create a new 3D box bi′=xi,yi,zi,hi+μ,wi+μ,li+μ,θi, so as to obtain additional context information, where xi,yi,zi is the center location of object bi, hi,wi,li is the size of object bi, θi is the object orientation of the bird’s view, and μ is a constant used to expand the box size. For each point, through the segmentation mask we perform an internal/external testing, to determine whether the point is within the expanded bounding box proposal bi′. If it is an internal point, the point and its features would be retained to refine the box bi. Finally, we will get 512 points for each candidate box and encode them to get the pooling feature Fp∈ℜ512×C1, where C1 represents the number of channels.


## 3.2.2. Spatial Shape Prediction

The foreground points of the candidate box constitute a shape describing semantic clues; however, this shape is usually incomplete. Therefore, based on the point completion framework PCN [23], we design a spatial structure prediction network to complete the missing part of the object in the candidate box. As shown in Figure 5, the network takes incomplete points as input and predicts the corresponding dense shape through the encoder-decoder. The encoder consists of two simple PointNet units (SharedMLP + Maxpool), each SharedMLP consisting of a 1×1 convolution layer, a BN layer, a ReLU layer, and a 1×1 convolution layer. The number of convolution output channels for the first SharedMLP is (128, 256), and the number of convolution output channels for the second SharedMLP is (512, 1024). The decoder consists of two stacked fully connected layers (Linear + BN + ReLU) and one fully connected layer (Linear), and the output is a 1024×3 matrix. The number of output channels for the three fully connected layers is (1024, 1024, 3∗1023). Unlike the coarse-to-fine pipeline in PCN, we believe that the coarse output is effective for subsequent processing, so we remove the fine output branch, thus saving GPU memory. In order to reduce the burden of training, we download the KITTI [23] car data set and trained our spatial shape prediction network in advance.Figure 5
Graphical spatial structure prediction network. The network takes the incomplete points as input and predicts the corresponding complete shape.Figure6 shows part of the visualization results of our spatial structure prediction model. It can be seen from the figure that our spatial structure prediction model performs well on automobiles and has a good generalization prospect.Figure 6
Visualization of spatial shape prediction of candidate objects. The original LiDAR point cloud (top) and the corresponding prediction results (bottom) for each object.


## 3.2.3. Structure Information Extraction and Fusion

To obtain the local and global context from the predicted spatial shapes, we use a PointNet++ [21] module to extract the structural information. First, we use the FPS algorithm to select 512 points from the predicted shape. Then for each point, we use the Ball Query algorithm to generate a local area. Finally, the PointNet units are applied to capture the local area feature C2 of each point, thereby obtaining the enhanced features Fs∈ℜ512×C2. In the refinement subnetwork part, we use a similar 3D box refinement network of PointRCNN [1] to further refine the box and confidence. The input of refining subnetwork consists of the canonical transformation coordinates of each pooling point, the pooling features, and the extracted spatial structure features. Since the pooling features and the spatial structure features come from different patterns, connecting them without any additional processing may cause interference. In order to better fusion spatial structure features and pooling features, we adopt the perspective-channel attention fusion [24] to obtain merged feature Fm∈ℜ512×C1+C2.


## 3.3. Loss Function

The proposed network is trained in an end-to-end manner. Our overall lossesLtotal include the two-stream RPN loss Lrpn in stage 1 and the box refining network loss Lrcnn in stage 2 as follows:(4)Ltotal=ωrpnLrpn+ωrcnnLrcnn,where ωrpn and ωrcnn are the coefficients that control the balance weight; we set the parameters ωrpn = 1.0 and ωrcnn = 1.0. Lrpn and Lrcnn adopt similar optimization objectives, including classification loss, regression loss, and consistency enhancement loss. For classification loss at the RPN stage, we use focal loss similar to [25] to balance positive and negative samples:(5)Lclsrpn=−1N∑i=1Nα1−y^iγlogy^i,ifyi=1,αyi+1−α1−yiy^iγlog1−y^i,otherwise,where yi is the target classification label, y^i is the positive sample prediction probability, N represents the number of targets, and α and γ are focal loss hyperparameters. For the regression loss in the RPN stage, we adopt a bin-based regression loss similar to [1] to regress the center point x,y,z, size l,h,w, and orientation θ:(6)Lresrpn=−1N∑i=1N∑u∈x,z,θFclsbin^u,binu+∑u∈x,y,z,h,w,l,θFregres^u,resu,where Fcls denotes the cross-entropy classification loss, Freg denotes the smooth 1 loss, binu and resu denote the bins and residuals of the ground truth, and bin^u and res^u denote the predicted bins and residuals of the ground truth. In addition, in order to improve the consistency of localization confidence and classification confidence, we add a consistency enhancement loss:(7)Lcerpn=−1N∑i=1Nlogci×Bi∩BigtBi∪Bigt,where Bi represents the predicted bounding box, Bigt represents the ground truth, and ci represents the classification confidence of the predicted box. In summary, Lrpn is a weighted sum of the three loss functions:(8)Lrpn=ωclsLclsrpn+ωregLregrpn+ωceLcerpn,where ωcls, ωreg, and ωce are used to control the balance coefficient of the importance degree of loss. We set the parameters ωcls = 1.0, ωreg = 1.0, and ωce = 5.0. Similarly, RCNN loss also includes classification loss, regression loss, and consistency enhancement loss. For RCNN classification loss, we adopt binary cross entropy loss:(9)Lclsrcnn=−1N∑i=1Nyilogy^i+1−yilog1−y^i,where yi is the target classification label, y^i is the target prediction probability, and N is the number of targets. RCNN regression loss Lregrcnn and consistency enhancement loss Lcercnn are defined in the same way as RPN. The weighted sum of the three loss functions of rcnn:(10)Lrcnn=ωclsLclsrcnn+ωregLregrcnn+ωceLcercnn.


## 4. Experiments

In this section, we evaluate our method on two common 3D object detection datasets, including the outdoor dataset KITTI [26] and the indoor dataset SUN-RGBD [27]. In Section 4.1, we introduce these datasets and evaluation metrics. In Section 4.2, we provide the implementation details of the experiment. In Section 4.3 and Section 4.4, we, respectively, show the comparison results of indoor and outdoor datasets. Finally, we conducted an extensive ablation study to analyze our proposed 3D target detection model in Section 4.5.


### 4.1. Datasets and Evaluation Metric

KITTI is the most popular standard benchmark dataset for autonomous driving, consisting of 7,481 samples for training and 7,518 samples for testing. As a common practice, the training samples are divided into a train set with 3,712 samples and a val set with 3,769 samples. The KITTI 3D object detection benchmark uses an average accuracy (AP) with a bounding box overlap of 0.7 as the evaluation indicator for cars, where three difficulty levels (easy, moderate, and hard) are taken into consideration. SUN-RGBD is a benchmark dataset for indoor 3D target detection. The dataset consists of 10,335 images and directional 3D bounding boxes with 37 target categories, including 5,285 images for training and 5,050 images for testing. We follow the same settings in VoteNet [28] and report the performance of 10 classes on SUN-RGBD. We use the average accuracy (AP) with a 3D overlap of 0.25 as the evaluation index of SUN-RGBD. We compare our method with the state-of-the-art methods in the KITTI and SUN-RGBD test set.


### 4.2. Implementation Details

Two-stream RPN takes LiDAR point clouds and camera images as input. We select 1,6384 points from the raw LiDAR point cloud as the input of the point stream and take the image with a resolution of 1280× 384 as the input of the image stream, which is the same as EPNet [15]. We use four SA layers (4096, 1024, 256, and 64) to subsample the input LiDAR point cloud and use four FP layers to recover the size of the point cloud for foreground segmentation and candidate box generation. Similarly, we use four convolutional blocks to downsample the input image and four transposed convolutional layers to restore the size of the image. In the NMS process, we select 8000 proposals generated by the two-stream RPN based on the classification confidence and then filter the redundant proposals with the NMS threshold of 0.8 to obtain 64 proposals for the refinement network. In the process of refining candidate boxes, we train a spatial structure prediction model in advance and then initialize the spatial shape prediction network with the weights. In the ablation experiment, we refer to the two-stage image classification strategy [29] to analyze the speed of our method. We train the model in an end-to-end manner on GeForce RTX 3090, the optimizer is ADAM [30], the initial learning rate is 0.002, and the weight attenuation is 0.001. The minibatch size is set to 2 and the model is trained for 40 epochs.


### 4.3. Experimental Results in KITTI

We compare the proposed two-stage detector with other state-of-the-art methods and submitted the results to the KITTI server for evaluation. As shown in Table1, we evaluate our method on the BEV detection benchmark and 3D object detection benchmark of the KITTI test data set. It can be seen that our method is significantly ahead of the advanced single-stage multisensor methods ContFuse [10], MAFF [31], MVX-Net [32], and MVAF-Net [11] in terms of 3D mAP by 10.32%, 5.64%, 4.18%, and 1.01%, respectively. It should be pointed out that our method is a point-based two-stage multisensor method, so we focus on the performance comparison with the point-based multisensor methods. It can be seen that our method outperforms all advanced point-based multisensor methods F-PointNet [12], IDMOD [33], PI-RCNN [14], and EPNet [15] by 10.84%, 5.45%, 5.29%, and 0.47%, respectively. At the same time, our method is also superior to most voxel-based methods.Table 1
Comparison with state-of-the-art methods on the KITTI test server.
TypeMethodModality3D detection (car)Bev detection (car)EasyMod.Hard3D mAPEasyMod.HardBev mAPStage 1SECOND [18]LiDAR84.6575.9668.7176.4491.8186.3781.0486.413DSSD [4]LiDAR88.3679.5774.5580.8392.6689.0285.8689.18MAFF [31]LiDAR & img.85.5275.0467.6176.0690.7987.3477.6685.26MVX-Net [32]LiDAR & img.85.9975.8670.7077.5291.8686.5381.4186.60MVAF-Net [11]LiDAR & img.87.8778.7175.4880.6991.9587.7385.0088.23Stage 2PointRCNN [1]LiDAR86.9675.6470.7077.7792.1387.3682.7287.41Fast PointRCNN [20]LiDAR85.2977.4070.2477.6490.8787.8480.5286.41Part-A2 [2]LiDAR87.8178.4973.5179.9491.7087.7984.6188.03F-PointNet [12]LiDAR & img.82.1969.7960.5970.8691.1784.6774.7784.54PI-RCNN [14]LiDAR & img.84.3774.8270.0376.4191.4485.8181.0086.08EPNet [15]LiDAR & img.89.8179.2874.5981.2394.2288.4783.6988.79IDMOD [33]LiDAR & img.84.5075.4168.8376.2589.4386.4678.9384.94F-PointPillars [34]LiDAR & img.88.9079.2878.0782.0890.2089.4388.7789.47OurLiDAR & img.89.9479.8975.2481.7094.3988.8484.3989.21The bold value indicates the highest performance.The visualization results of our method on KITTI are shown in Figure7. For better visualization, we project the 3D bounding box of LiDAR coordinates to the RGB image. The upper part is the image 3D detection result, and the lower part is the point cloud scene detection result. It can be seen that our method performs well in capturing distant cars, although these objects are difficult to identify in RGB images and are susceptible to sparse point clouds.Figure 7
Qualitative results of our method on the KITTI dataset. The detection results are shown in the image (upper) and the corresponding point cloud (lower).


### 4.4. Experimental Results in SUN-RGBD

We further perform experiments on SUN-RGBD data sets to verify the effectiveness of our method in indoor scenarios. Table2 shows the results compared with the most advanced methods. Our method achieves excellent detection performance, outperforming PointFusion [35], F-PointNet [12], VoteNet [28], MBDF-NET [36], and EPNet [15] by 16.1%, 6.2%, 2.5%, 0.7%, and 0.4%, respectively. Specifically, F-Pointnet and VoteNet both estimate 3D boundary boxes of point clouds based on 2D boundary box projections of images. PointFusion combines point cloud features and image features in a concatenation fashion. Different from them, our method establishes a correspondence between image features and point features, thus providing a clearer representation. In addition, comparing with multisensor-based methods, EPNet and MBDF-NET are particularly valuable. Because they also establish the mapping relationship between image features and point features, EPNet and MBDF-NET do not consider the point cloud sparse problem, and MBDF-NET is a three-branch detector.Table 2
Quantitative comparison with advanced methods on the SUN-RGBD test set.
MethodModalityBathtubBedBookshelfChairDeskDresserNightstandSofaTableToilet3D mAPPointFusion [35]L & I37.368.637.755.117.224.032.353.831.083.844.1F-PointNet [12]L & I43.381.133.364.224.732.058.161.151.190.954.0VoteNet [28]L74.483.028.875.322.029.862.264.047.390.157.7EPNet [15]L & I75.485.235.475.026.131.362.067.252.188.259.8MBDF-Net [36]L & I81.584.733.077.331.229.057.765.649.985.559.5OurL & I75.685.435.575.626.431.662.567.752.888.660.2L and I represent the LiDAR point cloud and camera image.The visualization results of our method on SUN-RGBD are shown in Figure8. Unlike the KITTI dataset, the SUN-RGBD dataset contains objects of multiple categories and different scales. It can be seen from Figure 8 that our method can better detect a variety of objects with obvious scale changes, including small objects (such as chair and dressing table) and large objects (such as sofa and bed).Figure 8
Qualitative results of our method on the SUN-RGND dataset. For each pair, the camera image is shown above and the corresponding point cloud detection result is shown below. The ground truth and detected boxes are highlighted with green and blue boxes, respectively.


### 4.5. Ablation Studies

We conduct a series of ablation studies on the KITTI dataset to analyze multiscale fusion RPN and spatial structure enhancement modules. All models are trained on the training set and evaluated on the validation set of the KITTI dataset for car detection. All evaluations on the validation set are conducted through 40 recall positions.


#### 4.5.1. Effect of Multiscale Fusion RPN

In Table3, we investigate the effectiveness of different structures in multiscale fusion RPN. We analyze the effect of each structure on stage 1 by removing one structure while leaving the others unchanged. To be fair for comparison, all the experiments shared the same fixed state 2. In the first row, we remove the image semantic branch, and the performance decreases significantly, which demonstrates the advantage of semantic segmentation. Then we compare two different fusion schemes. One is the single-scale feature propagation layer (SFP) fusion, which is similar to the multisensor feature fusion backbone network of EPNet [15], and the image semantic features are fused with the last feature propagation layer. The other is multiscale feature propagation layer (MFP) fusion, where image semantic features are fused with each feature propagation layer (see Figure 2). The results show that MFP is better than SFP by 0.25% in 3D mAP. This shows that the application of semantic features on multiscale feature propagation layer is effective. At the same time, we also give the inference time in Table 3. It can be seen that the inference time of SFP is similar to the baseline, and the time consumption of MFP does not increase much.Table 3
Effect of multiscale fusion module.
MethodEasyModerateHard3D mAPTime (ms)RPN baseline85.6676.4876.0579.4080SFP91.5982.3279.8984.693MFP92.282.4979.8784.85105SFP: single-scale feature propagation layer fusion; MFP: multiscale feature propagation layer fusion.


#### 4.5.2. Effect of Convolution Layer

Table4 shows the effects of different convolution layers on the performance of image semantic segmentation. We take the convolution layer number of Unet convolution block as the baseline. When the number of convolutional layers of the convolution block is increased appropriately, the AP is slightly increased, but excessively increasing the number of convolution layers of convolution blocks will reduce AP. This is because a reasonable depth of convolutional neural network can extract more image semantic features, but too deep network will lead to overfitting, which is not good for convergence. At the same time, it can be seen from Table 4 that the inference time increases slightly with the increase of the number of convolutional layers.Table 4
Performance with the number of different convolution layers.
LayerEasyModerateHard3D mAPTime (ms)292.282.4979.8784.85105392.2182.5480.2284.99119591.3882.3380.1984.63137


#### 4.5.3. Effect of Point Cloud Region Pooling

Table5 shows the effects of different pool context widths on performance. When no context information is pooled, the accuracy of 3D object detection, especially for those difficult instances, decreases significantly. Because the object might be obscured or far away from the sensor, difficult cases often have fewer points in the candidate box, which requires more contextual information to classify and refine the candidate box. As shown in Table 5, too large pooling context width can also result in performance drops because the pooled region of the current candidate box may include noisy foreground points for other objects.Table 5
Performance of point cloud region pooling with different context widths.
Context widthEasyModerateHard3D mAP092.1282.1279.6784.640.292.2182.5480.2284.990.491.7082.4579.9384.69


#### 4.5.4. Effect of Spatial Structure Enhancement

We explore the effects of the spatial information enhancement module in Table6. In the first row, we do not use the spatial information enhancement module. In the second row, we add the spatial information enhancement module and only use the simplest connection fusion, which reduces AP. This is because the pooling features and the spatial structure features come from different patterns, and connecting them without any additional processing produces interference. In the third row, we use perspective-channel attention fusion to fuse the spatial information enhancement module, and the gain of mAP is 0.42%. This is because the spatial information enhancement module promotes the model to better obtain spatial information. In addition, the inference time of our spatial information enhancement module is only increased by 11 ms compared with the RCNN baseline.Table 6
Effect of spatial information enhancement module.
MethodEasyModerateHard3D mAPTime (ms)RCNN baseline92.2182.5480.2284.99119SSE-con91.8582.2880.0084.71127SSE-att92.6183.0080.6385.41130SSE-con represents the spatial information enhancement module of simple connection. SSE-att represents the spatial information enhancement module based on attention fusion.


## 4.1. Datasets and Evaluation Metric

KITTI is the most popular standard benchmark dataset for autonomous driving, consisting of 7,481 samples for training and 7,518 samples for testing. As a common practice, the training samples are divided into a train set with 3,712 samples and a val set with 3,769 samples. The KITTI 3D object detection benchmark uses an average accuracy (AP) with a bounding box overlap of 0.7 as the evaluation indicator for cars, where three difficulty levels (easy, moderate, and hard) are taken into consideration. SUN-RGBD is a benchmark dataset for indoor 3D target detection. The dataset consists of 10,335 images and directional 3D bounding boxes with 37 target categories, including 5,285 images for training and 5,050 images for testing. We follow the same settings in VoteNet [28] and report the performance of 10 classes on SUN-RGBD. We use the average accuracy (AP) with a 3D overlap of 0.25 as the evaluation index of SUN-RGBD. We compare our method with the state-of-the-art methods in the KITTI and SUN-RGBD test set.


## 4.2. Implementation Details

Two-stream RPN takes LiDAR point clouds and camera images as input. We select 1,6384 points from the raw LiDAR point cloud as the input of the point stream and take the image with a resolution of 1280× 384 as the input of the image stream, which is the same as EPNet [15]. We use four SA layers (4096, 1024, 256, and 64) to subsample the input LiDAR point cloud and use four FP layers to recover the size of the point cloud for foreground segmentation and candidate box generation. Similarly, we use four convolutional blocks to downsample the input image and four transposed convolutional layers to restore the size of the image. In the NMS process, we select 8000 proposals generated by the two-stream RPN based on the classification confidence and then filter the redundant proposals with the NMS threshold of 0.8 to obtain 64 proposals for the refinement network. In the process of refining candidate boxes, we train a spatial structure prediction model in advance and then initialize the spatial shape prediction network with the weights. In the ablation experiment, we refer to the two-stage image classification strategy [29] to analyze the speed of our method. We train the model in an end-to-end manner on GeForce RTX 3090, the optimizer is ADAM [30], the initial learning rate is 0.002, and the weight attenuation is 0.001. The minibatch size is set to 2 and the model is trained for 40 epochs.


## 4.3. Experimental Results in KITTI

We compare the proposed two-stage detector with other state-of-the-art methods and submitted the results to the KITTI server for evaluation. As shown in Table1, we evaluate our method on the BEV detection benchmark and 3D object detection benchmark of the KITTI test data set. It can be seen that our method is significantly ahead of the advanced single-stage multisensor methods ContFuse [10], MAFF [31], MVX-Net [32], and MVAF-Net [11] in terms of 3D mAP by 10.32%, 5.64%, 4.18%, and 1.01%, respectively. It should be pointed out that our method is a point-based two-stage multisensor method, so we focus on the performance comparison with the point-based multisensor methods. It can be seen that our method outperforms all advanced point-based multisensor methods F-PointNet [12], IDMOD [33], PI-RCNN [14], and EPNet [15] by 10.84%, 5.45%, 5.29%, and 0.47%, respectively. At the same time, our method is also superior to most voxel-based methods.Table 1
Comparison with state-of-the-art methods on the KITTI test server.
TypeMethodModality3D detection (car)Bev detection (car)EasyMod.Hard3D mAPEasyMod.HardBev mAPStage 1SECOND [18]LiDAR84.6575.9668.7176.4491.8186.3781.0486.413DSSD [4]LiDAR88.3679.5774.5580.8392.6689.0285.8689.18MAFF [31]LiDAR & img.85.5275.0467.6176.0690.7987.3477.6685.26MVX-Net [32]LiDAR & img.85.9975.8670.7077.5291.8686.5381.4186.60MVAF-Net [11]LiDAR & img.87.8778.7175.4880.6991.9587.7385.0088.23Stage 2PointRCNN [1]LiDAR86.9675.6470.7077.7792.1387.3682.7287.41Fast PointRCNN [20]LiDAR85.2977.4070.2477.6490.8787.8480.5286.41Part-A2 [2]LiDAR87.8178.4973.5179.9491.7087.7984.6188.03F-PointNet [12]LiDAR & img.82.1969.7960.5970.8691.1784.6774.7784.54PI-RCNN [14]LiDAR & img.84.3774.8270.0376.4191.4485.8181.0086.08EPNet [15]LiDAR & img.89.8179.2874.5981.2394.2288.4783.6988.79IDMOD [33]LiDAR & img.84.5075.4168.8376.2589.4386.4678.9384.94F-PointPillars [34]LiDAR & img.88.9079.2878.0782.0890.2089.4388.7789.47OurLiDAR & img.89.9479.8975.2481.7094.3988.8484.3989.21The bold value indicates the highest performance.The visualization results of our method on KITTI are shown in Figure7. For better visualization, we project the 3D bounding box of LiDAR coordinates to the RGB image. The upper part is the image 3D detection result, and the lower part is the point cloud scene detection result. It can be seen that our method performs well in capturing distant cars, although these objects are difficult to identify in RGB images and are susceptible to sparse point clouds.Figure 7
Qualitative results of our method on the KITTI dataset. The detection results are shown in the image (upper) and the corresponding point cloud (lower).


## 4.4. Experimental Results in SUN-RGBD

We further perform experiments on SUN-RGBD data sets to verify the effectiveness of our method in indoor scenarios. Table2 shows the results compared with the most advanced methods. Our method achieves excellent detection performance, outperforming PointFusion [35], F-PointNet [12], VoteNet [28], MBDF-NET [36], and EPNet [15] by 16.1%, 6.2%, 2.5%, 0.7%, and 0.4%, respectively. Specifically, F-Pointnet and VoteNet both estimate 3D boundary boxes of point clouds based on 2D boundary box projections of images. PointFusion combines point cloud features and image features in a concatenation fashion. Different from them, our method establishes a correspondence between image features and point features, thus providing a clearer representation. In addition, comparing with multisensor-based methods, EPNet and MBDF-NET are particularly valuable. Because they also establish the mapping relationship between image features and point features, EPNet and MBDF-NET do not consider the point cloud sparse problem, and MBDF-NET is a three-branch detector.Table 2
Quantitative comparison with advanced methods on the SUN-RGBD test set.
MethodModalityBathtubBedBookshelfChairDeskDresserNightstandSofaTableToilet3D mAPPointFusion [35]L & I37.368.637.755.117.224.032.353.831.083.844.1F-PointNet [12]L & I43.381.133.364.224.732.058.161.151.190.954.0VoteNet [28]L74.483.028.875.322.029.862.264.047.390.157.7EPNet [15]L & I75.485.235.475.026.131.362.067.252.188.259.8MBDF-Net [36]L & I81.584.733.077.331.229.057.765.649.985.559.5OurL & I75.685.435.575.626.431.662.567.752.888.660.2L and I represent the LiDAR point cloud and camera image.The visualization results of our method on SUN-RGBD are shown in Figure8. Unlike the KITTI dataset, the SUN-RGBD dataset contains objects of multiple categories and different scales. It can be seen from Figure 8 that our method can better detect a variety of objects with obvious scale changes, including small objects (such as chair and dressing table) and large objects (such as sofa and bed).Figure 8
Qualitative results of our method on the SUN-RGND dataset. For each pair, the camera image is shown above and the corresponding point cloud detection result is shown below. The ground truth and detected boxes are highlighted with green and blue boxes, respectively.


## 4.5. Ablation Studies

We conduct a series of ablation studies on the KITTI dataset to analyze multiscale fusion RPN and spatial structure enhancement modules. All models are trained on the training set and evaluated on the validation set of the KITTI dataset for car detection. All evaluations on the validation set are conducted through 40 recall positions.


### 4.5.1. Effect of Multiscale Fusion RPN

In Table3, we investigate the effectiveness of different structures in multiscale fusion RPN. We analyze the effect of each structure on stage 1 by removing one structure while leaving the others unchanged. To be fair for comparison, all the experiments shared the same fixed state 2. In the first row, we remove the image semantic branch, and the performance decreases significantly, which demonstrates the advantage of semantic segmentation. Then we compare two different fusion schemes. One is the single-scale feature propagation layer (SFP) fusion, which is similar to the multisensor feature fusion backbone network of EPNet [15], and the image semantic features are fused with the last feature propagation layer. The other is multiscale feature propagation layer (MFP) fusion, where image semantic features are fused with each feature propagation layer (see Figure 2). The results show that MFP is better than SFP by 0.25% in 3D mAP. This shows that the application of semantic features on multiscale feature propagation layer is effective. At the same time, we also give the inference time in Table 3. It can be seen that the inference time of SFP is similar to the baseline, and the time consumption of MFP does not increase much.Table 3
Effect of multiscale fusion module.
MethodEasyModerateHard3D mAPTime (ms)RPN baseline85.6676.4876.0579.4080SFP91.5982.3279.8984.693MFP92.282.4979.8784.85105SFP: single-scale feature propagation layer fusion; MFP: multiscale feature propagation layer fusion.


### 4.5.2. Effect of Convolution Layer

Table4 shows the effects of different convolution layers on the performance of image semantic segmentation. We take the convolution layer number of Unet convolution block as the baseline. When the number of convolutional layers of the convolution block is increased appropriately, the AP is slightly increased, but excessively increasing the number of convolution layers of convolution blocks will reduce AP. This is because a reasonable depth of convolutional neural network can extract more image semantic features, but too deep network will lead to overfitting, which is not good for convergence. At the same time, it can be seen from Table 4 that the inference time increases slightly with the increase of the number of convolutional layers.Table 4
Performance with the number of different convolution layers.
LayerEasyModerateHard3D mAPTime (ms)292.282.4979.8784.85105392.2182.5480.2284.99119591.3882.3380.1984.63137


### 4.5.3. Effect of Point Cloud Region Pooling

Table5 shows the effects of different pool context widths on performance. When no context information is pooled, the accuracy of 3D object detection, especially for those difficult instances, decreases significantly. Because the object might be obscured or far away from the sensor, difficult cases often have fewer points in the candidate box, which requires more contextual information to classify and refine the candidate box. As shown in Table 5, too large pooling context width can also result in performance drops because the pooled region of the current candidate box may include noisy foreground points for other objects.Table 5
Performance of point cloud region pooling with different context widths.
Context widthEasyModerateHard3D mAP092.1282.1279.6784.640.292.2182.5480.2284.990.491.7082.4579.9384.69


### 4.5.4. Effect of Spatial Structure Enhancement

We explore the effects of the spatial information enhancement module in Table6. In the first row, we do not use the spatial information enhancement module. In the second row, we add the spatial information enhancement module and only use the simplest connection fusion, which reduces AP. This is because the pooling features and the spatial structure features come from different patterns, and connecting them without any additional processing produces interference. In the third row, we use perspective-channel attention fusion to fuse the spatial information enhancement module, and the gain of mAP is 0.42%. This is because the spatial information enhancement module promotes the model to better obtain spatial information. In addition, the inference time of our spatial information enhancement module is only increased by 11 ms compared with the RCNN baseline.Table 6
Effect of spatial information enhancement module.
MethodEasyModerateHard3D mAPTime (ms)RCNN baseline92.2182.5480.2284.99119SSE-con91.8582.2880.0084.71127SSE-att92.6183.0080.6385.41130SSE-con represents the spatial information enhancement module of simple connection. SSE-att represents the spatial information enhancement module based on attention fusion.


## 4.5.1. Effect of Multiscale Fusion RPN

In Table3, we investigate the effectiveness of different structures in multiscale fusion RPN. We analyze the effect of each structure on stage 1 by removing one structure while leaving the others unchanged. To be fair for comparison, all the experiments shared the same fixed state 2. In the first row, we remove the image semantic branch, and the performance decreases significantly, which demonstrates the advantage of semantic segmentation. Then we compare two different fusion schemes. One is the single-scale feature propagation layer (SFP) fusion, which is similar to the multisensor feature fusion backbone network of EPNet [15], and the image semantic features are fused with the last feature propagation layer. The other is multiscale feature propagation layer (MFP) fusion, where image semantic features are fused with each feature propagation layer (see Figure 2). The results show that MFP is better than SFP by 0.25% in 3D mAP. This shows that the application of semantic features on multiscale feature propagation layer is effective. At the same time, we also give the inference time in Table 3. It can be seen that the inference time of SFP is similar to the baseline, and the time consumption of MFP does not increase much.Table 3
Effect of multiscale fusion module.
MethodEasyModerateHard3D mAPTime (ms)RPN baseline85.6676.4876.0579.4080SFP91.5982.3279.8984.693MFP92.282.4979.8784.85105SFP: single-scale feature propagation layer fusion; MFP: multiscale feature propagation layer fusion.


## 4.5.2. Effect of Convolution Layer

Table4 shows the effects of different convolution layers on the performance of image semantic segmentation. We take the convolution layer number of Unet convolution block as the baseline. When the number of convolutional layers of the convolution block is increased appropriately, the AP is slightly increased, but excessively increasing the number of convolution layers of convolution blocks will reduce AP. This is because a reasonable depth of convolutional neural network can extract more image semantic features, but too deep network will lead to overfitting, which is not good for convergence. At the same time, it can be seen from Table 4 that the inference time increases slightly with the increase of the number of convolutional layers.Table 4
Performance with the number of different convolution layers.
LayerEasyModerateHard3D mAPTime (ms)292.282.4979.8784.85105392.2182.5480.2284.99119591.3882.3380.1984.63137


## 4.5.3. Effect of Point Cloud Region Pooling

Table5 shows the effects of different pool context widths on performance. When no context information is pooled, the accuracy of 3D object detection, especially for those difficult instances, decreases significantly. Because the object might be obscured or far away from the sensor, difficult cases often have fewer points in the candidate box, which requires more contextual information to classify and refine the candidate box. As shown in Table 5, too large pooling context width can also result in performance drops because the pooled region of the current candidate box may include noisy foreground points for other objects.Table 5
Performance of point cloud region pooling with different context widths.
Context widthEasyModerateHard3D mAP092.1282.1279.6784.640.292.2182.5480.2284.990.491.7082.4579.9384.69


## 4.5.4. Effect of Spatial Structure Enhancement

We explore the effects of the spatial information enhancement module in Table6. In the first row, we do not use the spatial information enhancement module. In the second row, we add the spatial information enhancement module and only use the simplest connection fusion, which reduces AP. This is because the pooling features and the spatial structure features come from different patterns, and connecting them without any additional processing produces interference. In the third row, we use perspective-channel attention fusion to fuse the spatial information enhancement module, and the gain of mAP is 0.42%. This is because the spatial information enhancement module promotes the model to better obtain spatial information. In addition, the inference time of our spatial information enhancement module is only increased by 11 ms compared with the RCNN baseline.Table 6
Effect of spatial information enhancement module.
MethodEasyModerateHard3D mAPTime (ms)RCNN baseline92.2182.5480.2284.99119SSE-con91.8582.2880.0084.71127SSE-att92.6183.0080.6385.41130SSE-con represents the spatial information enhancement module of simple connection. SSE-att represents the spatial information enhancement module based on attention fusion.


## 5. Conclusion

In this paper, we introduce a multiscale fusion RPN for features extraction and proposals generation. Besides, we also propose a novel spatial information enhancement module for detecting 3D objects from point clouds with the imbalanced density. Specifically, we design a spatial structure enhancement module to generate the complete shape of the candidate box and learn the structural information to enhance the features for box refinement. A large number of experiments verify the effectiveness of our proposed framework.



---

*Source: 4650660-2021-12-21.xml*