Next Article in Journal
A New Approach for Improving GNSS Geodetic Position by Reducing Residual Tropospheric Error (RTE) Based on Surface Meteorological Data
Previous Article in Journal
Merging Microwave, Optical, and Reanalysis Data for 1 Km Daily Soil Moisture by Triple Collocation
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

SPV-SSD: An Anchor-Free 3D Single-Stage Detector with Supervised-PointRendering and Visibility Representation

School of Automotive Studies, Tongji University, Shanghai 201804, China
*
Author to whom correspondence should be addressed.
Remote Sens. 2023, 15(1), 161; https://doi.org/10.3390/rs15010161
Submission received: 5 December 2022 / Revised: 19 December 2022 / Accepted: 21 December 2022 / Published: 27 December 2022

Abstract

:
Recently, 3D object detection based on multi-modal sensor fusion has been increasingly adopted in automated driving and robotics. For example, the semantic information provided by cameras and the geometric information provided by light detection and ranging (LiDAR) are fused to perceive 3D objects, as single modal sensors are unable to capture enough information from the environment. Many state-of-the-art methods fuse the signals sequentially for simplicity. By sequentially, we mean using the image semantic signals as auxiliary input for LiDAR-based object detectors would make the overall performance heavily rely on the semantic signals. Moreover, the error introduced by these signals may lead to detection errors. To remedy this dilemma, we propose an approach coined supervised-PointRendering to correct the potential errors in the image semantic segmentation results by training auxiliary tasks with fused features of the laser point geometry feature, the image semantic feature and a novel laser visibility feature. The laser visibility feature is obtained through the raycasting algorithm and is adopted to constrain the spatial distribution of fore- and background objects. Furthermore, we build an efficient anchor-free Single Stage Detector (SSD) powered by an advanced global-optimal label assignment to achieve a better time–accuracy balance. The new detection framework is evaluated on the extensively used KITTI and nuScenes datasets, manifesting the highest inference speed and at the same time outperforming most of the existing single-stage detectors with respect to the average precision.

Graphical Abstract

1. Introduction

Accurate and real-time 3D object detection is an indispensable task in the perception system of intelligent vehicles. Current LiDAR-only object detection approaches [1] have achieved impressive performance, especially with multi-views to take advantage of LiDAR point spatial information as much as possible. These include VISTA [2] based on both bird’s-eye view (BEV) and range view (RV) and MASS [3] using the dense top-view. However, these LiDAR-only methods still have some shortcomings. On the one hand, they rarely consider that the point cloud captured by LiDAR in the autonomous driving scene is not real “3D” data. Limited by the visibility constraints, the original point cloud cannot well perceive the information behind objects that LiDAR fails to detect. On the other hand, self-driving cars generally use multi-sensor combinations such as LiDAR and cameras: LiDAR provides accurate spatial information, while the camera provides rich context and semantic information; thus, data fusion can achieve the advantages of different types of complementary sensors.
The current fusion dilemma lies in the fact that most 3D detectors extract features on the bird’s-eye view (BEV), which is difficult to align with the front view image captured by the monocular camera. The sequential fusion style at the input stage can avoid this problem. A typical method is PointPainting [4]. It first projects the point cloud into the image semantic segmentation results, then appends the corresponding semantic scores to each point. This method enables easily missed small objects to be detected due to semantic information but decreases the precision of large objects which can be detected accurately by LiDAR-only methods (Table 1). A conclusion can be illustrated from the results reported on the KITTI benchmark [5] that PointPainting contributes to significant improvements on the average precision (AP) of pedestrians and cyclists, while contributing little to or even deteriorating the performance of car detection. The reason is that sequential fusion makes subsequent detection network rely heavily on the quality of semantic segmentation. As Figure 1 shows, many background points are misclassified as cars, inevitably leading to more false positives.
Current 3D object detectors often ignore the fact that measuring the 3D environment as (x,y,z) points destroys the hidden spatial distribution information [7,8]. According to the physical characteristics of raycasting, “visibility” ensures that there are no obstacles between the LiDAR origin and the detected object, and everything behind the detected object along its line-of-sight is occluded. The visibility constraint can estimate the free area distribution in the 3D space and can provide context information for object detection through modeling the bounding box coordinates as the Gaussian parameters (i.e., the mean and variance), and redesigning the loss function of the bounding box [9]. Moreover, the database sampling is one of the general augmentation strategies, which randomly copies and pastes virtual objects into point cloud scenes. However, such a method often inserts objects behind walls or buildings, ignoring their visibility and authenticity. In this paper, we adopt a visibility feature to assist both object detection and data augmentation.
Most voxel-based detectors prefer anchor-based detection heads. Anchor-based methods generate dense anchor boxes to guarantee the high recall. However, for complex scenes, such as in the nuScenes [10], anchor-based methods will generate dozens of anchors in consideration of the number of categories and orientation bins in each grid, resulting in cubic-level growth of parameters and slowing down the inference. In contrast, existing voxel-based anchor-free detectors directly regress bounding boxes in the foreground area of the heatmap, capable to run in real-time yet with limited performance. As shown in Table 2, the average precision of the car category achieved by the AFDet (anchor-free) on the KITTI validation dataset is 0.91% lower than the SECOND (anchor-based), and the gap between the CenterPoint and the SECOND is even larger.
To address the above issues, this paper explores an anchor-free 3D single-stage object detector with supervised-PointRendering and visibility representation. The supervised-PointRendering decorates points by appending image semantic segmentation results to points, with the additional point-wise supervision task to rectify incorrect semantic segmentation results. We adopt a raycasting algorithm to reconstruct the spatial visibility features of laser, then fuse them with the image semantic feature and point cloud feature. We also apply the optimal transport assignment strategy [15], which was originally used in the 2D object detection to boost the confidence-based global label assignment. We adapt it to the 3D object detection like [16,17], yet with modifications, such as adding orientation to the algorithm and enlarging center prior region of small objects to make it more suitable for 3D detection framework. With such an implementation, our anchor-free model can boost the inference speed while achieving a higher precision. Compared with existing approaches, the main highlights of this paper are summarized as follows:
  • We propose a novel supervised-PointRendering to eliminate the effects of incorrect image semantics on object detection, significantly improving the detection precision of all categories.
  • We introduce laser visibility features into the sequential fusion-based 3D object detection, which supplements more scene context to boost the 3D object detection.
  • We design an anchor-free detection head powered by the 3D optimal transport assignment (OTA-3D) in the voxel-based approaches, achieving a decent balance between precision and speed. At present, our model has achieved comparable precision with the state-of-the-art single-stage approaches on both KITTI and nuScenes datasets with an ultra-high inference speed, showing the generality and efficiency of our model in different traffic scenarios.
Figure 1. (a) Semantic segmentation results by DeeplabV3+ [18]; (b) the point cloud is painted with the 2D segmentation results. Cars are denoted in red. The zoomed figure depicts the misclassified ground area.
Figure 1. (a) Semantic segmentation results by DeeplabV3+ [18]; (b) the point cloud is painted with the 2D segmentation results. Cars are denoted in red. The zoomed figure depicts the misclassified ground area.
Remotesensing 15 00161 g001aRemotesensing 15 00161 g001b

2. Related Work

2.1. 3D Object Detection with Point Cloud

LiDAR-based 3D object detectors are generally divided into single-stage and two-stage detectors [19,20] depending on their architecture.
The single-stage detectors directly predict classification scores and object size and location in one stage. For instance, the VoxelNet [21] generates regularized data by voxelizing the point cloud, then feed the data into the standard 3D convolution neural network (CNN). SECOND [11] explores the sparse convolution to reduce the computation cost caused by standard 3D convolution and proposes the submanifold sparse convolution to prevent the feature diffusion. The Point-GNN [22] extracts features through a novel graph neural network and shows comparable performance with state-of-the-art two-stage detectors. The SASSD [23] employs an auxiliary network for box center regression and segmentation, aiming to learn the structure and localization information. This inspires us to exploit point-level supervision to eliminate the boundary-blurring effect in semantic segmentation.
Two-stage detectors typically refine the candidate proposals generated in the first stage. The Voxel-RCNN [24] applies a voxel-based CNN to generate regions of interest (RoIs) and refines the proposals by the devised voxel RoI pooling module. The PV-RCNN [25] takes advantage of the efficient voxel-based backbone network and the flexible PointNet-based set abstraction to extract more representative features. The Part- A 2 [26] learns intra-object part locations to enrich the RoI features, reducing the ambiguity of the bounding boxes.
It is worth noting that most voxel-based models use an anchor-based detection head. However, this mechanism has several drawbacks. First, to achieve the optimal detection performance, one needs to cluster a set of optimal anchors based on training data which requires heuristic tuning. Second, the anchor mechanism greatly increases both the number of predictions and the design complexity of detection heads. In contrast, the anchor-free mechanism reduces the parameter number of the detection head significantly, making the training and decoding phase of the detector much simpler. Some efforts have been made in voxel-based anchor-free detectors, such as the CenterPoint [14] and the AFDet [13]. However, these detectors achieve less outperforming results on the KITTI dataset than anchor-based detectors. In the 2D detection, anchor-free detectors [27,28,29] have developed rapidly in recent years and can exceed anchor-based detectors with suitable label assignments.
Given the high efficiency of the one-stage network and great potential of the anchor-free mechanism in 3D object detection, we focus on developing an anchor-free single-stage detection approach with improved precision, which has attained competitive performance with the top-performing single-stage detectors with a high inference speed.

2.2. Multi-Modal Fusion

Object detection methods utilizing LiDAR-camera fusion have progressed by leaps and bounds in recent years. According to [4], they can be divided into the following types: object-centric fusion, feature-level fusion, detection seeding and sequential fusion.
The representatives of object-centric fusion are the MV3D [30] and the AVOD [31]. They generate proposals from both the image and point cloud projection view, then perform deep feature fusion on the RoI features on the BEV or the front view. However, the projection from the point cloud to the image plane loses massive spatial information, which imposes an adverse effect on the detection accuracy.
The feature-level fusion, pioneered by the ContFuse [32], fuses features from the image and LiDAR backbone networks at different scales. These methods often calculate a mapping relationship to convert the point clouds to the image plane. The core problem lies in that each BEV feature vector of the point cloud can correspond to multiple pixels in the 2D image, resulting in fuzzy feature align and thus limited performance. To address this challenge, AutoAlign [33] proposes a cross-attention module to fit the mapping relationship. BEVFusion [34] accelerates the BEV pooling process during the view transformation process and achieves a decent balance between precision and speed.
The detection seeding methods such as the Frustrum PointNet [35] and the ConvNet [36] utilize 2D detection results to limit the frustum searching space to seed the 3D proposal. These detectors rely heavily on the performance of 2D detectors and thus heavily restrict the recall rate.
Sequential fusion is a simple, general yet effective strategy compared with other fusion methods. The LRPD [37] and the PointPainting [4] both use the output of an image semantic segmentation network to assist object detection. The PointPainting [4] projects LiDAR points into the semantic segmentation results of the image to decorate the points with semantic scores and then feed the painted points to the feature extraction network. However, the “boundary-blurring effect” occurs in the semantic segmentation process because the high-level feature map generally has quite low resolution. This error becomes more apparent when re-projecting them into the 3D space.

2.3. Visibility Representation

As far as we know, research focusing on spatial visibility representation is mostly carried out in robotic mapping. Buhmann et al. [38] used a 2D probabilistic occupancy map based on sonar sensor data to navigate mobile robots. Hornung et al. [39] propose a general 3D occupancy map to describe the space state, indicating the occupied, free and unknown area. The visibility representation through a raycasting algorithm is the core of constructing such occupancy maps.
Although visibility has gained popularity in the robotics area, the visibility reasoning has not received sufficient attention in 3D object detection. Richter et al. [40] integrated the occupancy grid map into the probabilistic framework to detect objects with known surface. Notably, Hu et al. [41] proposed the concept that LiDAR point clouds are not real “3D” but “2.5D”. They reconstructed the spatial visibility state through a raycasting algorithm and converted 3D spatial features to 2D multi-channel feature maps, which were then integrated into the PointPillars [12]. Such visibility features can be directly concatenated with a voxelized point cloud and thus bring better data alignment.

3. Proposed Method

In this part, we devise the anchor-free 3D single-stage object detector with supervised-PointRendering and LiDAR visibility. Section 3.1 presents our detection network architecture. Section 3.2 describes the supervised-PointRendering to enhance the point cloud feature. Section 3.3 describes the extraction of the spatial visibility feature of laser to provide scene representation. Section 3.4 introduces the anchor-free detection head with an improved label assignment. Section 3.5 presents the losses used in training.

3.1. Network Architecture

Our framework is composed of data processing, a backbone network, and a detection head.
In the data processing, we adopt an off-the-shelf image segmentor to output the per pixel semantic class scores and then append the scores to the points. We reconstruct the spatial visibility feature though a raycasting algorithm and concatenate them with the voxelized point clouds. The painted point clouds with visibility features are fed into the backbone network.
Our backbone network contains two modules: a typical 3D convolution backbone network followed by a spatial semantic feature aggregation (SSFA) module [42] for feature extraction and an auxiliary network which aims to exploit point-wise supervisions. As manifested in Figure 2, the backbone network is made up of four convolution blocks, each of which contains submanifold convolutions with a kernel size of three. In the last three blocks, the features are downsampled with a stride of two. Then, we concatenate features from the backbone along the height dimension to generate the BEV feature maps. High-level semantic features and low-level spatial features are fused adaptively in the semantic aggregation. The auxiliary network scatters the convolution features back to points and performs foreground segmentation and center regression tasks.
The anchor-free detection head has three tasks, including classification, localization and intersection-over-union (IoU) prediction with a confidence function [43] to rectify the confidence used in the non-maximum-suppression (NMS) post-processing. In the end, we employ a global optimal label assignment to find the best label assigning at minimal global costs.

3.2. Supervised-PointRendering

3.2.1. PointRendering

With an input image, the image semantic segmentation network outputs semantic class scores per pixel. The high-level features of an image are represented by these scores. We transform LiDAR points into the camera coordinate system, project them into the image, and append the segmentation scores of corresponding pixels to the LiDAR point features. We choose segmentation scores as image feature encoding because scores implicitly contain uncertainty information, which requires the network to distinguish the semantic category itself, thus reducing the influence of incorrect semantic segmentation on subsequent networks.

3.2.2. Point-Wise Supervision Task

As Figure 2 shows, we utilize the point-wise foreground segmentation to rectify the incorrect segmentation results. Through the inverse process of voxelization, we convert the voxel feature to the world coordinate system. The feature from N 1 points are propagated to N points by interpolation, where N 1 indicates the number of new points converted by voxels, while N denotes the number of the original points. In the interpolation approach, the feature vector f ( x j ) at each coordinate of original points x j can be calculated by a weighted average on features of its k nearest neighbors x i among the new points (k is empirically set to three), interpreted by
f ( x j ) = i = 1 k w i j ( x i ) f ( x i ) i = 1 k w ( x i ) , j = 1 , , C ,
where the weight w i j ( x i ) is the inverse square of distance between the original point x j and the new point x i :
w i j ( x i ) = 1 x j x i 2 .
Instead of directly concatenating the obtained features with the next-stage features [23], our method applies a shared multi-layer perception (MLP) to process the features to yield point feature encoding h x i , which is then concatenated with next-stage features, enriching the encoding for the supervision task. We apply 1 × 1 convolutions to generate predictions for the foreground segmentation task. Moreover, we add a center regression task to predict offsets from object points to the corresponding instance center, aiming to guide the backbone network to learn object structure information. These tasks are detachable in the inference stage, introducing no extra computational cost.
The point-wise supervision loss includes the foreground segmentation loss and the center regression loss, which are, respectively, interpreted by the focal loss and the smooth-L1 loss. The foreground segmentation label is a binary value with one to indicate that the point is within the bounding box of ground truth. The center regression predicts the offsets of foreground points to the corresponding object centers.

3.3. Visibility Representation

3.3.1. Ray Casting Algorithm

The raycasting algorithm in a horizontal plane is depicted in Figure 3. Given the center coordinates of the current voxel ( x , y ) , ( t m x , t m y ) are the time intervals from the current position to the boundary of the adjacent voxel along the ray in the x and y direction, respectively. By comparing the time intervals in two directions, one can know whether the ray reaches the horizontal face or the vertical face first. We traverse along the ray to reach the next voxel, update ( x , y ) and ( t m x , t m y ) and mark the traversed grid. The algorithm can be easily extended to 3D space by adding corresponding variables along the z axis. Specifically, the algorithm calculates the arrival time at six faces of the current voxel to check their intersection with the exiting ray. It then proceeds to the adjacent voxel with shared face. Each ray starts at the voxel at the sensor origin and iterates the above calculation until it reaches the voxel (precomputed) occupied by the LiDAR points. Such an algorithm is very efficient because of the linear calculation.

3.3.2. Spatial Visibility States

A grid can be assigned with one of the three visibility states: unknown, occupied and free space, which are represented by specific numerical values. Here, we first initialize all voxels as unknown, then execute the raycasting algorithm for each laser ray. The raycasting algorithm draws a line from the LiDAR sensor origin and the object 3D point. The voxels which the line passes are considered as free-space, and the last voxel that encloses the LiDAR point is regarded as occupied. If one grid cell is traversed by multiple rays with one indicating it is occupied and others indicating they are free, they are regarded as occupied under such circumstances. Details about this procedure can be referred to Algorithm 1. Figure 4 depicts the visibility representation of laser rays. It can be seen that the spatial visibility information is consistent with the original point cloud, providing strong support for the quantitative results obtained in the experiment.
Algorithm 1 Vanilla voxel traversal algorithm
Require: 
lidar origin o , raw point cloud P r a w , voxelized painted points P v o x , ending voxel v e .
Ensure: 
occupancy grids O , voxelized points with visibility states P v i s
1:
O [:]←UNKNOWN;
2:
for p in P r a w  do
3:
    v ← i n i t i a l _ v o x e l ( p ) ;
4:
    while v ≠ v e  do
5:
        v ← n e x t _ v o x e l ( v , p o ) ;
6:
        if v = v e  then
7:
            O [v] ← OCCUPIED;
8:
           break;
9:
        else
10:
            O [v] ← FREE;
11:
        end if
12:
    end while
13:
     P v i s c o n c a t ( P v o x , O ) ;
14:
end for

3.4. Anchor-Free Detection Head with OTA-3D

3.4.1. OTA-3D

We adapt the optimal transport assignment (OTA) strategy [15] from 2D object detection to 3D detection. Different from 2D methods, we adjust the candidate area for objects of different size flexibly and introduce rotated-IoU loss to the global cost. We first regard the predictions from the grids which are within a fixed region, e.g., 0.8 m × 0.8 m around the center of car and cyclist, and 1.6 m × 1.6 m around the pedestrian, as positive candidates. Then, we finely select the global-optimal positive samples from these positive candidates. The steps are depicted in Figure 5 and introduced as follows.
  • Calculating the cost matrix: Its element indicates the pair-wise matching cost between a positive candidate box and a ground truth. Different from the 2D OTA, the pair-wise cost c i j is composed of the classification loss L i j c l s , the bounding box regression loss L i j r and the BEV rotated-IoU loss L i j I o U between a positive candidate i and the ground truth box j, which is calculated as:
    c i j = L i j c l s + L i j r + λ L i j I o U .
  • Calculating the dynamic k: For each ground truth, its IoU (intersection over union) score is calculated with all positive candidates. We select the top N box predictions and sum their IoU scores as k, which is further rounded by the floor operation. Here, we set N = 10 ; k is thus the number of positive samples assigned to one ground truth bounding box.
  • Selecting the top k predictions with the least cost.
  • Filtering the repeated predictions: In the case that the same prediction matches multiple ground truth bounding boxes, the prediction is assigned to the ground truth bounding box with the least cost.
Finally, the corresponding grids of those positive predictions are assigned as positives, while the rest of the grids are considered as negatives.

3.4.2. Anchor-Free Detection Head

To achieve real-time performance without decreasing precision, we build an anchor-free detection head with OTA-3D.
In this detection head, we reduce the predictions for each location from N c l a s s × N o r i groups to one group. Each location directly predicts its class and offsets to the instance center, as well as the normalized distance from the mean size of the predicted class. For regression targets, we apply the following box encodings:
x t = x g x a d a , y t = y g y a d a , z t = z g z a h a , w t = log w g w a , l t = log l g l a , h t = log h g h a ,
in which x, y and z indicate the center coordinates; w, l and h denote the width, length and height, respectively; the subscripts t, a and g represent the encoded value, the mean size of each class and the corresponding ground truth, respectively; and d a = ( l a ) 2 + ( w a ) 2 denotes the diagonal of the mean size box.
Given that the orientation regression is difficult without prior information, we apply a hybrid formulation of classification and regression to predict the orientation. Specifically, we split 2 π into N a bins. The network predicts both the angle bin and the residuals relative to the bin. In the experiment, we set N a = 12 .
In order to mitigate the misalignment between classification confidence and localization accuracy, we add a branch to predict IoU score [42] to rectify the confidence in the non-maximum-suppression (NMS) process. The confidence is estimated as C = c · i β , where C is the confidence in the NMS process, i denotes the predicted IoU, c denotes the classification score, and β is a hyper-parameter that controls the distinction between accurate and inaccurate predictions. It is set to four.

3.5. Loss Function

The overall loss L includes the detection head loss L h e a d and the point-wise supervision loss for foreground segmentation L s e g and center regression L c t r :
L = L h e a d + ω L s e g + μ L c t r .
Here we set ω = 0.9, μ = 2.0 to balance the point-wise supervision task from the main task. The detection head loss includes the classification loss L c , the bounding box regression loss L r and the 3D D-IoU loss L I o U [44]. Among them, the 3D D-IoU is calculated by:
D I o U = I o U ρ 2 b , b g t c 2
where b, b g t represent the centers of the predicted bounding box and the ground truth, respectively, ρ represents their Euclidean distance, and c represents the diagonal distance of the smallest closed region that contains both the predicted bounding box and its ground truth. The IoU prediction loss L h e a d is calculated as follows:
L h e a d = 1 N c i L c ( s i , u i ) + λ 1 1 N p i [ u i > 0 ] L r + λ 2 1 N p L I o U + λ 3 1 N p L c o n f ,
where λ i is the weight of each task; N c and N p , respectively, denote the number of total proposals and positive proposals selected by the OTA-3D; s i is the classification score for each grid, while u i denotes the corresponding class label. Here, we set λ 1 = 1.0 , λ 2 = 5.0 and λ 3 = 1.0 . The 3D D-IoU loss is based on [45], and L c o n f is calculated by the smooth-L1 loss. For classification, the focal loss is chosen, which is:
L c l s = α ( 1 p a ) γ log p a ,
where p a denotes the estimated probability, and α and γ are hyper-parameters set to 0.25 and 2, following the work [46].
The center regression loss L d i s t , the size regression loss L s i z e and the angle regression loss L a n g l e make up the regression loss L r . The L d i s t regularizes offsets from positive grid location to the corresponding instance centers. The targets for L s i z e are the offsets between the real object size and the average size of its category. We adopt the smooth-L1 loss for both L d i s t and L s i z e . Considering that the angle localization loss cannot distinguish flipped boxes, our angle loss contains the orientation classification loss L o r i _ c l s and the corresponding residual prediction loss L o r i _ r e g , which can be interpreted as:
L a n g l e = L o r i _ c l s ( d c a , t c a ) + L o r i _ r e g ( d r a , t r a ) ,
where d c a is the predicted orientation, d r a is its residual while t c a , and t r a are their ground truths.

4. Experiments and Discussion

4.1. Datasets

Our method is evaluated on both the KITTI [5] and nuScenes [10] datasets.
The KITTI detection dataset includes 7481 images for training and 7518 images for testing. Furthermorfe, the training data are split into two subsets, a training set with 3712 frames and a validation set with 3769 frames. We evaluate our method on all three classes, namely car, pedestrian and cyclist, and our evaluation criterion is chosen as the average precision (AP) with 40 recall positions. All of these positions follow the IoU threshold of 0.7 for Car and 0.5 for Pedestrian and Cyclist. In addition, according to the object size, occlusion state and truncation level, the dataset is divided into three difficulty levels: easy, moderate and hard.
The nuScenes dataset [10] contains 1000 scenes, each consisting of a video sequence. For each video, the 360 views are annotated only for the key frames (every 0.5 s). The dataset is officially divided into subsets for training, validation and testing. The training dataset includes 700 scenes (28,130 frames), while the subset for validation and testing contains 150 scenes (6019 frames) and 150 scenes (6008 frames), correspondingly. The annotations comprised ten classes. The corresponding RGB images that cover the 360 field-of-view are also provided for each key frame. We choose the mean Average Precision (mAP) and the nuScenes detection score (NDS) as the main metrics; mAP calculates the center offsets on the BEV plane as the evaluation standard instead of the intersection ratio (IoU). NDS is a weighted sum of mAP and the mean average errors on size (mASE), translation (mATE), orientation (mAOE), attribute (mAAE) and velocity (mAVE).

4.2. Setup of Supervised-PointRendering

4.2.1. Setup on KITTI Dataset

As for the semantic segmentation network, we pre-trained the DeeplabV3+ [18] on the Cityscapes [47] dataset. Note that in the Cityscapes semantic segmentation benchmark, the “rider” (considered as “pedestrian”) and the “bicycle” are two different categories, while in the KITTI object detection they are grouped into one category “cyclist”. We solve this problem by mapping a “bicycle” and a “pedestrian”, which is within a distance of 1m to the bike, into a “cyclist”, and classifying the “bicycle” without people around it as the “background”. After PointRendering, the input dimension of the point cloud has changed from four to eight, namely ( x , y , z , r , s b g , s c a r , s p e d e , s c y c ) , where r is the intensity, and s represents the segmentation score of each class.

4.2.2. Setup on nuScenes Dataset

We choose the HTCNet [48] as the image semantic segmentation network due to its outstanding performance and pretrain it on the nuImage [49] dataset. Remarkably, on the nuScenes dataset, the LiDAR and camera work at different frequencies. Therefore, we adopt the transform procedure from [10]. During the image projection, there are laser points projecting on two images simultaneously in the case that the field of view of two cameras are overlapped. We adopt the average segmentation score of the overlapped images to decorate the points. For sweeps that do not have synchronous images, we assign them with images which are adjacent to the capture time of these sweeps.

4.3. Setup of Visibility Representation

4.3.1. Visibility Consistency in Data Augmentation

To keep the consistency of the spatial visibility state after the database sampling of data augmentation, we adopt the“drilling” strategy [41], which allows the rays between the scene origin and added targets to traverse “occupied” voxels in the original scene. As illustrated in Figure 6, for the authenticity of the scene, part of the wall is “drilled” by removing the orange points of the wall.

4.3.2. Spatial Visibility Feature Extraction

Numerical values of the three occupancy states are empirically defined as: unknown (0.5), occupied (0.7) and free space (0.4). Given one point cloud frame as input, the LiDAR origin is set to (0, 0, 0). If the input consists of aggregated LiDAR sweeps, the LiDAR origin is changed according to the position of the current frame relative to the previous frame, and Bayesian filtering is applied to accumulate these temporal visibility states as a 3D probability occupancy map.
The visibility computation is implemented in C++ and integrated into PyTorch training. We use the multi-threading and SIMD instructions to speed up the parallel execution, making the computation much faster. On two Intel(R) Xeon(R) Platinum 8163 CPU, it takes 18.2 ± 2.1 ms on average to compute the visibility features.

4.4. Choice of Key Parameter

We give an explanation about the choice of two typical parameters.
One is about the choice of visibility encoding. The reconstructed space visibility is represented by three different states, namely, unknown, occupied and free. For these three states, we use three numerical values to encode them. In this paper, we consider the following two groups of parameter settings to represent these states.
  • Setting 1: The unknown state is set to 0, the occupied state is set to 1 and the free state is set to −1.
  • Setting 2: According to the formula provided by Octomap [39], the unknown state is set to 0.5, the occupied state is set to 0.7 and the free state is set to 0.4.
We tested these two settings on the KITTI validation set, and the results of pedestrian detection are shown in Table 3. Notably, here we only demonstrate the experiment results on pedestrians because the visibility feature improves the precision of pedestrians more obviously. According to the final performance, [0.7, 0.5, 0.4] is finally selected as the visibility encoding parameters in our model.
Another is about the choice of the value of β in the confidence function. We follow the same setting in the CIA-SSD [42] which obtains the best result with β = 4.
For the KITTI dataset, we, respectively, set [0, 70.4] m, [−40, 40] m and [−3, 1] m as the detection range on the x, y and z axes. The size of the input voxel is set as 0.05 × 0.05 × 0.1 m 3 , and we use the ADAM optimizer to train the network with an initial learning rate of 0.003. The network is trained with a batch size of 56 on 8 RTX 2080 Ti GPUs, and the learning rate is decayed by 10 at 35 epochs and 40 epochs.
For the nuScenes dataset, the detection range on the x, y and z axes is set to [−54, 54] m, [−54, 54] m, [−5, 3] m, respectively. We set the input voxel size as 0.75 × 0.75 × 0.2 m 3 and train the entire network with a batch size 12 for 20 epochs on 4 NVIDIA GeForce RTX 3090. We follow the same training schedule as [14], and two test-time augmentations are adopted, including double flip testing and yaw rotations.
Here, we decouple the semantic segmentation network and the spatial visibility calculation from the detection framework for a better storage management and inference speed measurement.

4.5. Comparison with State of the Art

4.5.1. Results on the KITTI dataset

As revealed in Table 4, our model outperforms other advanced strategies on the difficulty levels of “moderate” and “hard” and on the classes of car and cyclist by a large margin. Specifically in the car category, our method outperforms PointPillars, SECOND and VoxelNet, respectively, by 6.03%, 4.38% and 15.23% on the moderate AP. What is worth mentioning is that our model surpasses the existing anchor-free model, the CenterPoint, by 6.38% on the moderate AP. Compared to the very recent methods, i.e., the SA-SSD [23] and the CIA-SSD [42], our model achieves a gain of about 1.24% and 2.53% on the hard AP, showing the effectiveness of our strategy.
The results demonstrate that our model deals fairly well with difficult objects, which have more occlusions or sparser points. Moreover, the accurate image semantic segmentation results provided by the supervised-PointRendering effectively eliminate the false negative predictions caused by sparse points and occlusions. In addition, the inference speed of our model surpasses all state-of-the-art voxel-based single stage detectors (with preprocessed image semantic segmentation). The FPS results in Table 4 are from the official KITTI leader board. Compared with other one-stage models, the high efficiency of our model is mainly attributed to the anchor-free detection head, which reduces a large number of parameters and computation cost.
However, our mAP on pedestrians underperforms the state-of-the-art methods on the test set. One of the reasons is that the number of pedestrians in KITTI is much less than that of cars. Even with database sampling in data augmentation, the variety of pedestrians is still lacking. Additionally, the voxel resolution of a pedestrian on the BEV is much worse than that of a cyclist or a car (see Figure 7). Due to the above issues, it is difficult to learn a pedestrian detector well with voxelized features on the KITTI dataset. A similar problem also appears in the VoxelNet [24].
Table 5 shows the comparison between our method and other advanced single-stage methods. It can be seen that the direct PointPainting decreases the precision of car detection, while our supervised-PointRendering successfully solves this problem, with gains of (1.73%, 6.49% and 10.39%) on three difficulty levels in the car category compared with the SECOND baseline. Remarkably, our method also surpasses the SA-SSD and CIA-SSD by 3.06% and 3.16% on the “moderate” difficulty level of car.

4.5.2. Results on nuScenes Dataset

Given the fact that the nuScenes dataset provides more classes in different scales and its point clouds are sparser, a modified version of our method is presented. In this version, we predict the centers of the objects by adopting a class-specific center heatmap head, and we utilize regression heads to estimate other location information such as size, rotation and velocity following [14]. The results (Table 6) on the test set show that our network achieves remarkable performance, validating the generality of our model on the different driving platforms. Our model achieves an improvement of 4.1% and 7.3% on the NDS and mAP metrics over the CenterPoint, showing the effectiveness of our fusion strategy and extra visibilty feature. Specifically, it improves the AP on traffic cone (Tr. Cone), Trailer and Motorcycle (Motor.) over 10%. Our results on the validation set (Table 7) show that our model surpasses the FUTR3D [51] by 0.5% and 0.7% on Car and Bus. Notably, our model demonstrates strong competitiveness in the detection of difficult object classes such as construction vehicle (Cons. Veh.) and pedestrian.

4.6. Ablation Study

In this part, we first study the effectiveness of the anchor-free detection head with OTA-3D. Then, we analyse the effect of supervised-PointRendering and the spatial visibility feature comprehensively. We conduct the experiments on the KITTI validation set.

4.6.1. Study on Anchor-Free Detection Head with OTA-3D

We adopt the backbone from the SECOND network [11] combined with the 3D D-IoU loss and an IoU prediction branch. We re-implement the 3D IoU loss based on [45] and update it to the 3D D-IoU loss. Then, we add an IoU prediction branch to further improve the AP of each category. The results show that these strategies benefit large objects such as cars.
Table 8 shows that our anchor-free head with OTA-3D boosts the moderate AP of car by 3.29%. The combination of anchor-free head and suitable label assignment strategy not only reduces the parameter amount to 1 N c l a s s × N o r i , but also yields better performance compared to anchor-based approaches such as SA-SSD [23] and CIA-SSD [42]. Our baseline for the subsequent ablation experiments is the model with the above strategies.

4.6.2. Study on Supervised-PointRendering

This experiment compares different semantic encodings, such as the object category ID, segmentation score, one-hot encoding and VoxelRendering. The VoxelRendering is to voxelize the segmentation scores, extract semantic features through 3D sparse convolution and fuse them with point cloud features before downsampling. Among them, the segmentation score encoding attains the largest gain in AP, which is 0.9% higher than that with numerical number encoding in the car category (Table 9). We argue that the segmentation score implies the classification confidence information, guiding the model to distinguish the category itself.
It can be seen in Table 9 that the AP on car drops after pure PointRendering. After utilizing point-wise supervision, our model boosts the moderate AP by 1.86%, indicating that the foreground segmentation task corrects the segmentation errors successfully. For pedestrian and cyclist, supervised-PointRendering even improves their performance by 3.94% and 0.04%, showing the importance of accurate image semantics in small object detection.
In Figure 8, qualitative examples are also presented to illustrate the benefit of our supervised-PointRendering. It can be seen that the SECOND with PointRendering alone misses more cars, while our method avoids this phenomenon and achieves the best detection performance.

4.6.3. Study on Spatial Visibility Fusion

Here we compare fusion of spatial visibility feature at different stages, as previously shown in Figure 2. Results in Table 10 show that the early fusion outperforms the late fusion, indicating that the early fusion is more conducive to data alignment. After supplementing the spatial visibility information, the performance of detecting small objects improves significantly. The moderate APs of pedestrian and cyclist increase by 3.07% and 1.41%, respectively. Since the “Easy” and “Moderate” difficulty levels in KITTI represent the situation that can be completely observed or less occluded, it is consistent with the meaning of “visibility” in principle.
However, by employing both spatial visibility and supervised-PointRendering, the AP on pedestrian drops a little. Since the spatial visibility is calculated based on voxels, we assume that it is a similar problem to that in Section 4.5.1. Therefore, the learning of spatial visibility negatively impacts the performance of supervised point-pointing on pedestrian detection.

4.6.4. Run Time and Computation Efficiency

In this section, we analyze the inference speed and time consumption of each component. We first compare our model with five popular single-stage detectors, and the result is shown in Table 11. It can be seen that our model is the fastest among them, with an average inference time of only 30.33 ms. Then, we split the pipeline into the following components: (1) point cloud pre-processing, (2) point cloud voxelization, (3) network forward processing, and (4) post-processing by NMS. As shown in Table 12, compared with the baseline method SECOND, Our anchor-free head has greatly decreased the inference time of the network, specifically, saving ∼9.4ms in model forward step. Moreover, due to the inference time of semantic segmentation network, our data reading step is a little bit slow (0.1 ms slower than SECOND).
We also analyze the computation efficiency of our model. We compare the FLOPs (floating point operations) and parameters in Table 13. The results show that the parameter amount and calculation cost of our model are rather small, demonstrating the high efficiency of our model.

5. Conclusions and Future Work

In this paper, we address the limitations of current LiDAR-only methods by presenting an anchor-free detection framework with a novel sensor fusion method and the LiDAR visibility representation. We propose the approach of supervised-PointRendering, which uses point-wise supervision to eliminate the influence of erroneous boundary segmentation on large object detection, improving the precision of all categories by a large margin. Current 3D object detectors process 3D point data with their coordinates yet ignore their hidden information about the free space. Here, we introduce the spatial visibility features of LiDAR to provide more spatial information for object detection. Through combining an anchor-free head with suitable label assignment, our detector solves the bottleneck of the inference time. By the experiment results on the KITTI and the nuScenes datasets, we demonstrate that our model achieves performance comparable to state-of-the-art single-stage methods with an ultra-high inference speed. The test results also demonstrate that our detector has strong robustness against different traffic scenarios.
Since this paper focuses on the improvement of 3D object detection, the adopted semantic segmentation approach is less optimized. In future work, the inference speed of the semantic segmentation network will be further promoted. Moreover, we will focus on exploiting the potential of a multi-level camera-LiDAR fusion model to boost the performance of 3D object detection.

Author Contributions

Conceptualization, W.T.; methodology, L.Y. and W.T.; software, L.Y. and L.W.; validation, L.Y.; formal analysis, L.Y.; data curation, L.W.; writing—original draft preparation, L.Y. and L.W.; writing—review and editing, W.T. and Z.W.; visualization, L.W.; supervision, W.T. and Z.Y.; project administration, W.T.; funding acquisition, W.T. All authors have read and agreed to the published version of the manuscript.

Funding

Project supported by the National Natural Science Foundation of China (No.52002285), the Shanghai Science and Technology Commission (No.21ZR1467400), the original research project of Tongji University (No.22120220593), and the National Key R&D Program of China (No.2021YFB2501104).

Data Availability Statement

The datasets generated and analysed during the current study are available in the [KITTI] repository and the [nuScenes] repository. [http://www.cvlibs.net/datasets/kitti (accessed on 4 December 2022)] and [https://www.nuscenes.org/nuscenes (accessed on 4 December 2022)].

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Wang, B.; Lan, J.; Gao, J. LiDAR filtering in 3D object detection based on improved RANSAC. Remote Sens. 2022, 14, 2110. [Google Scholar] [CrossRef]
  2. Deng, S.; Liang, Z.; Sun, L.; Jia, K. VISTA: Boosting 3D Object Detection via Dual Cross-VIew SpaTial Attention. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, New Orleans, LA, USA, 19–24 June 2022; pp. 8448–8457. [Google Scholar]
  3. Peng, K.; Fei, J.; Yang, K.; Roitberg, A.; Zhang, J.; Bieder, F.; Heidenreich, P.; Stiller, C.; Stiefelhagen, R. MASS: Multi-attentional semantic segmentation of LiDAR data for dense top-view understanding. IEEE Trans. Intell. Transp. Syst. 2022, 23, 15824–15840. [Google Scholar] [CrossRef]
  4. Vora, S.; Lang, A.H.; Helou, B.; Beijbom, O. PointPainting: Sequential Fusion for 3D Object Detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 4603–4611. [Google Scholar]
  5. Geiger, A.; Lenz, P.; Urtasun, R. Are we ready for autonomous driving? The KITTI vision benchmark suite. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Providence, RI, USA, 16–21 June 2012; pp. 3354–3361. [Google Scholar]
  6. Shi, S.; Wang, X.; Li, H. PointRCNN: 3D Object Proposal Generation and Detection From Point Cloud. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 16–20 June 2019; pp. 770–779. [Google Scholar]
  7. Kamal, A.; Dhakal, P.; Javaid, A.Y.; Devabhaktuni, V.K.; Kaur, D.; Zaientz, J.; Marinier, R. Recent advances and challenges in uncertainty visualization: A survey. J. Vis. 2021, 24, 861–890. [Google Scholar] [CrossRef]
  8. Yang, L.; Hyde, D.; Grujic, O.; Scheidt, C.; Caers, J. Assessing and visualizing uncertainty of 3D geological surfaces using level sets with stochastic motion. Comput. Geosci. 2019, 122, 54–67. [Google Scholar] [CrossRef]
  9. Choi, J.; Chun, D.; Kim, H.; Lee, H.J. Gaussian yolov3: An accurate and fast object detector using localization uncertainty for autonomous driving. In Proceedings of the IEEE/CVF International Conference on Computer Vision, Seoul, Republic of Korea, 27 October–2 November 2019; pp. 502–511. [Google Scholar]
  10. Caesar, H.; Bankiti, V.; Lang, A.H.; Vora, S.; Liong, V.E.; Xu, Q.; Krishnan, A.; Pan, Y.; Baldan, G.; Beijbom, O. nuScenes: A Multimodal Dataset for Autonomous Driving. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 11618–11628. [Google Scholar]
  11. Yan, Y.; Mao, Y.; Li, B. SECOND: Sparsely Embedded Convolutional Detection. Sensors 2018, 18, 3337. [Google Scholar] [CrossRef] [Green Version]
  12. Lang, A.H.; Vora, S.; Caesar, H.; Zhou, L.; Yang, J.; Beijbom, O. PointPillars: Fast Encoders for Object Detection From Point Clouds. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 16–20 June 2019; pp. 12689–12697. [Google Scholar]
  13. Ge, R.; Ding, Z.; Hu, Y.; Wang, Y.; Chen, S.; Huang, L.; Li, Y. AFDet: Anchor Free One Stage 3D Object Detection. arXiv 2020, arXiv:2006.12671. [Google Scholar]
  14. Yin, T.; Zhou, X.; Krähenbühl, P. Center-based 3D Object Detection and Tracking. In Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Nashville, TN, USA, 19–25 June 2021; pp. 11779–11788. [Google Scholar]
  15. Ge, Z.; Liu, S.; Li, Z.; Yoshie, O.; Sun, J. OTA: Optimal Transport Assignment for Object Detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Nashville, TN, USA, 19–25 June 2021; pp. 303–312. [Google Scholar]
  16. Yang, L.; Hou, W.; Cui, C.; Cui, J. GOSIM: A multi-scale iterative multiple-point statistics algorithm with global optimization. Comput. Geosci. 2016, 89, 57–70. [Google Scholar] [CrossRef]
  17. Chen, Y.; Tai, L.; Sun, K.; Li, M. Monopair: Monocular 3d object detection using pairwise spatial relationships. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 14–19 June 2020; pp. 12093–12102. [Google Scholar]
  18. Chen, L.C.; Zhu, Y.; Papandreou, G.; Schroff, F.; Adam, H. Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation. arXiv 2018, arXiv:1802.02611. [Google Scholar]
  19. Zhao, Z.Q.; Zheng, P.; Xu, S.t.; Wu, X. Object detection with deep learning: A review. IEEE Trans. Neural Netw. Learn. Syst. 2019, 30, 3212–3232. [Google Scholar] [CrossRef] [Green Version]
  20. Zou, Z.; Shi, Z.; Guo, Y.; Ye, J. Object detection in 20 years: A survey. arXiv 2019, arXiv:1905.05055. [Google Scholar]
  21. Zhou, Y.; Tuzel, O. VoxelNet: End-to-End Learning for Point Cloud Based 3D Object Detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–22 June 2018; pp. 4490–4499. [Google Scholar]
  22. Shi, W.; Rajkumar, R.R. Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloud. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 1708–1716. [Google Scholar]
  23. He, C.H.; Zeng, H.; Huang, J.; Hua, X.; Zhang, L. Structure Aware Single-Stage 3D Object Detection From Point Cloud. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 11870–11879. [Google Scholar]
  24. Deng, J.; Shi, S.; Li, P.; Zhou, W.; Zhang, Y.; Li, H. Voxel R-CNN: Towards High Performance Voxel-based 3D Object Detection. In Proceedings of the AAAI, Virtual, 2–9 February 2021. [Google Scholar]
  25. Shi, S.; Guo, C.; Jiang, L.; Wang, Z.; Shi, J.; Wang, X.; Li, H. PV-RCNN: Point-Voxel Feature Set Abstraction for 3D Object Detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 10526–10535. [Google Scholar]
  26. Shi, S.; Wang, Z.; Wang, X.; Li, H. Part-A2 Net: 3D Part-Aware and Aggregation Neural Network for Object Detection from Point Cloud. arXiv 2019, arXiv:1907.03670. [Google Scholar]
  27. Redmon, J.; Farhadi, A. YOLO9000: Better, Faster, Stronger. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21–26 July 2017; pp. 6517–6525. [Google Scholar]
  28. Law, H.; Deng, J. CornerNet: Detecting Objects as Paired Keypoints. In Proceedings of the European Conference on Computer Vision (ECCV), Munich, Germany, 8–14 September 2018. [Google Scholar]
  29. Zhou, X.; Wang, D.; Krähenbühl, P. Objects as Points. arXiv 2019, arXiv:1904.07850. [Google Scholar]
  30. Chen, X.; Ma, H.; Wan, J.; Li, B.; Xia, T. Multi-view 3D Object Detection Network for Autonomous Driving. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA, 21–26 July 2017; pp. 6526–6534. [Google Scholar]
  31. Ku, J.; Mozifian, M.; Lee, J.; Harakeh, A.; Waslander, S.L. Joint 3D Proposal Generation and Object Detection from View Aggregation. In Proceedings of the IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), Madrid, Spain, 1–5 October 2018; pp. 1–8. [Google Scholar]
  32. Liang, M.; Yang, B.; Wang, S.; Urtasun, R. Deep Continuous Fusion for Multi-sensor 3D Object Detection. In Proceedings of the ECCV, Munich, Germany, 8–14 September 2018. [Google Scholar]
  33. Chen, Z.; Li, Z.; Zhang, S.; Fang, L.; Jiang, Q.; Zhao, F.; Zhou, B.; Zhao, H. AutoAlign: Pixel-Instance Feature Aggregation for Multi-Modal 3D Object Detection. arXiv 2022, arXiv:2201.06493. [Google Scholar]
  34. Liu, Z.; Tang, H.; Amini, A.; Yang, X.; Mao, H.; Rus, D.; Han, S. BEVFusion: Multi-Task Multi-Sensor Fusion with Unified Bird’s-Eye View Representation. arXiv 2022, arXiv:2205.13542. [Google Scholar]
  35. Qi, C.; Liu, W.; Wu, C.; Su, H.; Guibas, L.J. Frustum PointNets for 3D Object Detection from RGB-D Data. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, 18–22 June 2018; pp. 918–927. [Google Scholar]
  36. Wang, Z.; Jia, K. Frustum ConvNet: Sliding Frustums to Aggregate Local Point-Wise Features for Amodal. In Proceedings of the IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), Macau, China, 3–8 November 2019; pp. 1742–1749. [Google Scholar]
  37. Fürst, M.; Wasenmüller, O.; Stricker, D. LRPD: Long Range 3D Pedestrian Detection Leveraging Specific Strengths of LiDAR and RGB. In Proceedings of the IEEE 23rd International Conference on Intelligent Transportation Systems (ITSC), Rhodes, Greece, 20–23 September 2020; pp. 1–7. [Google Scholar]
  38. Buhmann, J.M.; Burgard, W.; Cremers, A.B.; Fox, D.; Hofmann, T.; Schneider, F.E.; Strikos, J.; Thrun, S. The Mobile Robot Rhino. In Proceedings of the SNN Symposium on Neural Networks, Nijmegen, The Netherlands, 14–15 September 1995. [Google Scholar]
  39. Hornung, A.; Wurm, K.M.; Bennewitz, M.; Stachniss, C.; Burgard, W. OctoMap: An efficient probabilistic 3D mapping framework based on octrees. Auton. Robot. 2013, 34, 189–206. [Google Scholar] [CrossRef] [Green Version]
  40. Richter, S.; Wirges, S.; Königshof, H.; Stiller, C. Fusion of range measurements and semantic estimates in an evidential framework / Fusion von Distanzmessungen und semantischen Größen im Rahmen der Evidenztheorie. TM-Tech. Mess. 2019, 86, 102–106. [Google Scholar] [CrossRef]
  41. Hu, P.; Ziglar, J.; Held, D.; Ramanan, D. What You See is What You Get: Exploiting Visibility for 3D Object Detection. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 10998–11006. [Google Scholar]
  42. Zheng, W.; Tang, W.; Chen, S.; Jiang, L.; Fu, C.W. CIA-SSD: Confident IoU-Aware Single-Stage Object Detector From Point Cloud. In Proceedings of the AAAI, Virtual, 2–9 February 2021. [Google Scholar]
  43. Bodla, N.; Singh, B.; Chellappa, R.; Davis, L.S. Soft-NMS — Improving Object Detection with One Line of Code. In Proceedings of the IEEE International Conference on Computer Vision (ICCV), Venice, Italy, 22–29 October 2017; pp. 5562–5570. [Google Scholar]
  44. Zheng, Z.; Wang, P.; Liu, W.; Li, J.; Ye, R.; Ren, D. Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression. In Proceedings of the AAAI, New York, NY, USA, 7–12 February 2020. [Google Scholar]
  45. Zhou, D.; Fang, J.; Song, X.; Guan, C.; Yin, J.; Dai, Y.; Yang, R. IoU Loss for 2D/3D Object Detection. In Proceedings of the International Conference on 3D Vision (3DV), Quebec City, QC, Canada, 16–19 September 2019; pp. 85–94. [Google Scholar]
  46. Lin, T.Y.; Goyal, P.; Girshick, R.B.; He, K.; Dollár, P. Focal Loss for Dense Object Detection. IEEE Trans. Pattern Anal. Mach. Intell. 2020, 42, 318–327. [Google Scholar] [CrossRef]
  47. Cordts, M.; Omran, M.; Ramos, S.; Rehfeld, T.; Enzweiler, M.; Benenson, R.; Franke, U.; Roth, S.; Schiele, B. The Cityscapes Dataset for Semantic Urban Scene Understanding. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA, 27–30 June 2016; pp. 3213–3223. [Google Scholar]
  48. Chen, K.; Pang, J.; Wang, J.; Xiong, Y.; Li, X.; Sun, S.; Feng, W.; Liu, Z.; Shi, J.; Ouyang, W.; et al. Hybrid Task Cascade for Instance Segmentation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, 15–20 June 2019; pp. 4969–4978. [Google Scholar]
  49. nuImages. 2020. Available online: https://www.nuscenes.org/nuimages (accessed on 4 December 2022).
  50. Pang, S.; Morris, D.D.; Radha, H. CLOCs: Camera-LiDAR Object Candidates Fusion for 3D Object Detection. In Proceedings of the 2020 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), Las Vegas, NV, USA, 25–29 October 2020; pp. 10386–10393. [Google Scholar]
  51. Chen, X.; Zhang, T.; Wang, Y.; Wang, Y.; Zhao, H. Futr3d: A unified sensor fusion framework for 3d detection. arXiv 2022, arXiv:2203.10642. [Google Scholar]
  52. Yin, J.; Shen, J.; Guan, C.; Zhou, D.; Yang, R. Lidar-based online 3d video object detection with graph-based message passing and spatiotemporal transformer attention. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA, 14–19 June 2020; pp. 11495–11504. [Google Scholar]
  53. Yang, Z.; Sun, Y.; Liu, S.; Jia, J. 3DSSD: Point-Based 3D Single Stage Object Detector. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA, 14–19 June 2020; pp. 11037–11045. [Google Scholar]
  54. Zhu, X.; Zhou, H.; Wang, T.; Hong, F.; Li, W.; Ma, Y.; Li, H.; Yang, R.; Lin, D. Cylindrical and asymmetrical 3d convolution networks for lidar-based perception. IEEE Trans. Pattern Anal. Mach. Intell. 2022, 44, 6807–6822. [Google Scholar] [CrossRef]
  55. Zhu, B.; Jiang, Z.; Zhou, X.; Li, Z.; Yu, G. Class-balanced grouping and sampling for point cloud 3D object detection. arXiv 2019, arXiv:1908.09492. [Google Scholar]
  56. Chen, Q.; Sun, L.; Cheung, E.; Yuille, A.L. Every view counts: Cross-view consistency in 3D object detection with hybrid-cylindrical-spherical voxelization. Adv. Neural Inf. Process. Syst. 2020, 33, 21224–21235. [Google Scholar]
Figure 2. The pipeline of our proposed anchor-free 3D single-stage object detection network with supervised-PointRendering and spatial visibility representation.
Figure 2. The pipeline of our proposed anchor-free 3D single-stage object detection network with supervised-PointRendering and spatial visibility representation.
Remotesensing 15 00161 g002
Figure 3. Ray traversal on the 2D grids. The letters a–f represent the orders of grids which the ray reaches.
Figure 3. Ray traversal on the 2D grids. The letters a–f represent the orders of grids which the ray reaches.
Remotesensing 15 00161 g003
Figure 4. (a) A single sweep of LiDAR and the corresponding single-layer instantaneous visibility. The denotation of three visibility states are: occupied-space (red), unknown-space (light blue) and free-space (dark blue); (b) aggregated LiDAR sweeps and their single-layer superimposed temporal occupancy states. Color with increased saturation indicates a greater probability of occupancy state of the corresponding voxel.
Figure 4. (a) A single sweep of LiDAR and the corresponding single-layer instantaneous visibility. The denotation of three visibility states are: occupied-space (red), unknown-space (light blue) and free-space (dark blue); (b) aggregated LiDAR sweeps and their single-layer superimposed temporal occupancy states. Color with increased saturation indicates a greater probability of occupancy state of the corresponding voxel.
Remotesensing 15 00161 g004aRemotesensing 15 00161 g004b
Figure 5. Optimal transport assignment in 3D object detection.
Figure 5. Optimal transport assignment in 3D object detection.
Remotesensing 15 00161 g005
Figure 6. The “drilling” strategy in data augmentation through visibility reasoning: (a) the raw point cloud; (b) the point cloud after ground truth database sampling. The orange part will be removed after “drilling”.
Figure 6. The “drilling” strategy in data augmentation through visibility reasoning: (a) the raw point cloud; (b) the point cloud after ground truth database sampling. The orange part will be removed after “drilling”.
Remotesensing 15 00161 g006
Figure 7. Qualitative results on the KITTI dataset. The pink, blue and red boxes represent cars, cyclists and pedestrians, respectively.
Figure 7. Qualitative results on the KITTI dataset. The pink, blue and red boxes represent cars, cyclists and pedestrians, respectively.
Remotesensing 15 00161 g007
Figure 8. From top to bottom are ground truths, detection results of the SECOND, the painted SECOND and our SPV-SSD on the KITTI dataset. The ground truths and predictions are marked in blue and green respectively.
Figure 8. From top to bottom are ground truths, detection results of the SECOND, the painted SECOND and our SPV-SSD on the KITTI dataset. The ground truths and predictions are marked in blue and green respectively.
Remotesensing 15 00161 g008
Table 1. Deterioration of car detection performance by sequentially appending image semantic segmentation results to point cloud. Test results of Painted PointPillars are obtained on the validation set using the official implementation, while the test results of Painted PointRCNN are from the KITTI server.
Table 1. Deterioration of car detection performance by sequentially appending image semantic segmentation results to point cloud. Test results of Painted PointPillars are obtained on the validation set using the official implementation, while the test results of Painted PointRCNN are from the KITTI server.
MethodCar (%)Pedestrian (%)Cyclist (%)
EasyMod.HardEasyMod.HardEasyMod.Hard
PointPillars [4]87.2276.9573.5257.7552.2947.9182.2963.2659.82
Painted
PointPillars [4]
86.2676.7770.2561.5056.1550.0379.1264.1860.79
Delta−0.96−0.18−3.27+3.75+3.86+2.12−3.17+0.92+0.97
PointRCNN [6]86.9675.6470.7047.9839.3736.0174.9658.8252.53
Painted
PointRCNN [4]
82.1171.7067.0850.3240.9737.8777.6363.7855.89
Delta−4.85−3.94−3.62+2.34+1.6+1.86+2.67+4.96+3.36
Table 2. Performance comparison of existing voxel-based anchor-free/-based detectors on the KITTI dataset. Above are the validation results of car detection by the SECOND, the PointPillars and the AFDet approach. The AFDet only provides results for the KITTI validation dataset. Below are the car detection results of the SECOND, the PointPillars and the CenterPoint approach on the KITTI test dataset.
Table 2. Performance comparison of existing voxel-based anchor-free/-based detectors on the KITTI dataset. Above are the validation results of car detection by the SECOND, the PointPillars and the AFDet approach. The AFDet only provides results for the KITTI validation dataset. Below are the car detection results of the SECOND, the PointPillars and the CenterPoint approach on the KITTI test dataset.
MethodDet. Head3D AP (%)BEV AP (%)
EasyMod.HardEasyMod.Hard
SECOND [11]anchor-based87.4376.4869.1089.9687.0779.66
PointPillars [12]anchor-based83.7376.0469.1289.6886.3484.38
AFDet [13]anchor-free85.6875.5769.3189.4285.4580.56
SECONDanchor-based84.6575.9668.7189.3983.7778.59
PointPillarsanchor-based82.5874.3168.9990.0796.5682.81
CenterPoint [14]anchor-free81.1773.9669.4888.4785.0581.19
Table 3. Performance comparison of visibility state encoding for pedestrian detection on the KITTI validation dataset. Here, the baseline is tested with the integration of visibility features.
Table 3. Performance comparison of visibility state encoding for pedestrian detection on the KITTI validation dataset. Here, the baseline is tested with the integration of visibility features.
Visibility Encoding [U, O, F]Pedestrian AP (BEV) (%)mAP (BEV) (%)Pedestrian AP (3D) (%)mAP (3D) (%)
EasyMod.HardEasyMod.Hard
[1, 0, −1]62.8658.0852.8857.9458.7655.9047.7454.13
[0.7, 0.5, 0.4]64.5760.1554.5059.7460.8457.9849.7756.20
Delta+1.71+2.07+1.62+1.80+2.08+2.08+2.03+2.07
Table 4. The 3D detection AP of compared methods on the KITTI test data. The modalities include lidar (L) and images (I). The CLOCs_SecCas is short for the approach of SECOND+Cascaded R-CNN.
Table 4. The 3D detection AP of compared methods on the KITTI test data. The modalities include lidar (L) and images (I). The CLOCs_SecCas is short for the approach of SECOND+Cascaded R-CNN.
Method ModalityCar (%)Pedestrian (%)Cyclist (%) FPS
EasyMod.HardEasyMod.HardEasyMod.Hard
Voxelnet [21] L77.4765.1157.7339.4833.6931.5161.2248.3644.374.4
SECOND [11] L84.6575.9668.7145.3135.5233.1475.8360.8253.6720
PointPillars [12]L82.5874.3168.9951.4541.9238.8977.1058.6551.9242
PointRCNN [6]L86.9675.6470.7047.9839.3736.0174.9658.8252.53-
CenterPoint [14]L81.1773.9669.4847.2539.2836.7873.0456.6750.60-
CLOCs_SecCas [50]L86.3878.4572.45 - - - - - - -
SA-SSD [23]L88.7579.7974.16 - - - - - - 25
CIA-SSD [42]L89.5980.2872.87 - - - - - - 32
MV3D [30]L & I74.9763.6354.00 - - - - - - 2.8
AVOD-FPN [31]L & I83.0771.7665.7350.4642.2739.0463.7650.5544.9310
F-PointNet [35]L & I82.1969.7960.5950.5342.1538.0872.2756.1249.015.9
OursL & I87.2280.3475.4045.8338.4536.0378.3664.4056.9233
Table 5. The 3D detection AP of compared single-stage methods on the KITTI validation set. The AP is calculated with 11 recall points.
Table 5. The 3D detection AP of compared single-stage methods on the KITTI validation set. The AP is calculated with 11 recall points.
MethodCar (%)Pedestrian (%)Cyclist (%)
EasyMod.HardEasyMod.HardEasyMod.Hard
PointPillars [12]87.2276.9573.5257.7552.2947.9182.2963.2659.82
Painted
PointPillars [4]
86.2676.7770.2561.5056.1550.0379.1264.1860.79
Delta−0.96−0.18−3.27+3.75+3.86+2.12−3.17+0.92+0.97
VoxelNet [24]81.9765.4662.8557.8653.4248.8767.1747.6545.11
SECOND [11]87.4376.4869.10------
SA-SSD [23]90.1579.9178.78------
CIA-SSD [42]90.0479.8178.80------
AFDet [13]85.6875.5769.31------
Ours89.1682.9779.4960.7555.6750.2083.6769.5964.21
Table 6. Comparison with state-of-the-art methods on the nuScenes test set. The table is mainly sorted by nuScenes detection score (NDS).
Table 6. Comparison with state-of-the-art methods on the nuScenes test set. The table is mainly sorted by nuScenes detection score (NDS).
MethodsStagesNDSmAPCarTruckBusTrailerCons. Veh.Ped.Motor.BicycleTr.ConeBarrier
WYSIWYG [41]One41.935.079.130.446.640.17.165.018.20.128.834.7
PointPillars [12]One45.330.568.423.028.223.44.159.727.41.130.838.9
3DVID [52]One53.145.479.733.647.143.118.176.540.77.958.848.8
3DSSD [53]One56.442.681.247.261.430.512.670.236.08.631.147.9
Cylinder3D [54]One61.650.6----------
CenterPoint [14]Two65.558.084.651.060.253.217.583.453.728.776.770.9
CGBS [55]One63.352.881.148.554.942.910.580.151.522.370.965.7
CVCNet [56]One64.255.882.649.559.451.116.283.061.838.869.769.7
OursOne69.665.386.656.565.860.429.786.568.342.980.675.5
Table 7. Comparison with state-of-the-art methods on the nuScenes validation dataset. The results of PointPillars, SECOND and FUTR3D are reproduced with their official implementation.
Table 7. Comparison with state-of-the-art methods on the nuScenes validation dataset. The results of PointPillars, SECOND and FUTR3D are reproduced with their official implementation.
MethodsStagesNDSmAPCarTruckBusTrailerCons. Veh.Ped.Motor.BicycleTr. ConeBarrier
PointPillars [12]One45.329.670.525.034.520.04.559.916.81.729.633.2
SECOND [11]One48.427.175.521.929.013.00.459.916.90.022.532.2
MEGVII [55]One62.550.781.651.767.237.514.877.742.617.457.459.2
FUTR3D [51] One 68.0 64.2 86.3 61.5 71.9 42.1 26.0 82.6 73.663.3 70.1 64.4
OursOne69.264.786.861.072.642.127.186.571.856.273.369.7
Table 8. Comparison of different strategies for the detection head. OTA-3D Head denotes the anchor-free head with the OTA-3D strategy. The AP is calculated with 11 recall points.
Table 8. Comparison of different strategies for the detection head. OTA-3D Head denotes the anchor-free head with the OTA-3D strategy. The AP is calculated with 11 recall points.
OTA-3D Head3D D-IoUIoU Pred.Car (%)Pedestrian (%)Cyclist (%)
Mod.Mod.Mod.
76.4851.1466.74
77.3752.6867.23
78.5654.5567.65
81.8553.7068.13
Table 9. Comparison of different semantic encodings. The class_id, one-hot, VR, seg. score and SUPV denote the object category ID, one-hot encoding, VoxelRendering, segmentation score and supervised-PointRendering, respectively. The AP is calculated with 11 recall points.
Table 9. Comparison of different semantic encodings. The class_id, one-hot, VR, seg. score and SUPV denote the object category ID, one-hot encoding, VoxelRendering, segmentation score and supervised-PointRendering, respectively. The AP is calculated with 11 recall points.
Class_IDOne-HotVRSeg. ScoreSUPVCar (%)Ped. (%)Cyc. (%)
Mod.Mod.Mod.
81.8553.7068.13
78.4557.3664.48
77.9255.5965.12
79.1054.1964.25
79.3555.1067.20
83.2157.6468.17
Table 10. Comparison of spatial visibility fusion strategies; vis.(early) indicates visibility feature in early fusion while vis.(late) is for late fusion. SUPV is short for the supervised-PointRendering. The AP is calculated with 11 recall points.
Table 10. Comparison of spatial visibility fusion strategies; vis.(early) indicates visibility feature in early fusion while vis.(late) is for late fusion. SUPV is short for the supervised-PointRendering. The AP is calculated with 11 recall points.
Vis.(Early) Vis.(Late) SUPVCar (%)Ped (%)Cyc. (%)
Mod.Mod.Mod.
81.8553.7068.13
79.3357.9869.54
78.2655.9169.03
82.9755.6769.59
Table 11. Comparison of our model with other single-stage detectors on run time performance. Results are in milliseconds.
Table 11. Comparison of our model with other single-stage detectors on run time performance. Results are in milliseconds.
1-StagePoint-GNNAssociate-3DdetSASSD3DSSDTANetOurs (1-Stage)
time (ms)6436040.13834.7530.33
Table 12. Run time analysis (in milliseconds) of different steps during inference.
Table 12. Run time analysis (in milliseconds) of different steps during inference.
MethodsPre-Proc.Voxel.Net ForwardNMSOverall
SECOND [11]1.56.637.50.746.3
SA-SSD [23]1.5<0.0137.90.740.1
Ours1.6<0.0128.10.630.3
Table 13. Network parameters (M) and FLOPs (G) of our model and other models.
Table 13. Network parameters (M) and FLOPs (G) of our model and other models.
MethodsParameters (M)FLOPs (G)
SECOND [11]5.3376.84
PV-RCNN [25]13.13178.50
Ours3.7666.45
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Yin, L.; Tian, W.; Wang, L.; Wang, Z.; Yu, Z. SPV-SSD: An Anchor-Free 3D Single-Stage Detector with Supervised-PointRendering and Visibility Representation. Remote Sens. 2023, 15, 161. https://doi.org/10.3390/rs15010161

AMA Style

Yin L, Tian W, Wang L, Wang Z, Yu Z. SPV-SSD: An Anchor-Free 3D Single-Stage Detector with Supervised-PointRendering and Visibility Representation. Remote Sensing. 2023; 15(1):161. https://doi.org/10.3390/rs15010161

Chicago/Turabian Style

Yin, Lingmei, Wei Tian, Ling Wang, Zhiang Wang, and Zhuoping Yu. 2023. "SPV-SSD: An Anchor-Free 3D Single-Stage Detector with Supervised-PointRendering and Visibility Representation" Remote Sensing 15, no. 1: 161. https://doi.org/10.3390/rs15010161

APA Style

Yin, L., Tian, W., Wang, L., Wang, Z., & Yu, Z. (2023). SPV-SSD: An Anchor-Free 3D Single-Stage Detector with Supervised-PointRendering and Visibility Representation. Remote Sensing, 15(1), 161. https://doi.org/10.3390/rs15010161

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop