S-N E RF: N EURAL R ADIANCE F IELDS FOR S TREET
V IEWS
Anonymous authors
Paper under double-blind review
A BSTRACT
Neural Radiance Fields (NeRFs) aim to synthesize novel views of objects and scenes, given the object-centric camera views with large overlaps.
However, we conjugate that this paradigm does not fit the nature of the street views that are collected by many self-driving cars from the large-scale unbounded scenes.
Also, the onboard cameras perceive scenes without much overlapping.
Thus, existing
NeRFs often produce blurs, “floaters” and other artifacts on street-view synthesis.
In this paper, we propose a new street-view NeRF ( S-NeRF ) that considers novel view synthesis of both the large-scale background scenes and the foreground moving vehicles jointly.
Specifically, we improve the scene parameterization function and the camera poses for learning better neural representations from street views.
We also use the the noisy and sparse
LiDAR points to boost the training and learn a robust geometry and reprojection based confidence to address the depth outliers.
Moreover, we extend our S-NeRF for reconstructing moving vehicles that is impracticable for conventional NeRFs.
Thorough experiments on the large-scale driving datasets ( e.g. , nuScenes and Waymo) demonstrate that our method beats the state-of-the-art rivals by reducing 7 ∼ 40% of the mean-squared error in the street-view synthesis and a 45% PSNR gain for the moving vehicles rendering.
I NTRODUCTION
Neural Radiance Fields (Mildenhall et al., 2020) have shown impressive performance on photorealistic novel view rendering.
However, original NeRFs (Mildenhall et al., 2020; Deng et al., 2022) are usually designed for object-centric scenes and require camera views to be heavily overlapped (as shown in Figure 1(a)).
Recently, more and more street view data are collected by self-driving cars.
The reconstruction and novel view rendering for street views can be very useful in driving simulation, data generation, AR and VR.
However, these data are often collected in the unbounded outdoor scenes ( e.g. nuScenes (Caesar et al., 2019) and Waymo (Sun et al., 2020) datasets).
The camera placements of such data acquisition systems are usually in a panoramic settings without object-centric camera views (Figure 1(b)).
Moreover, the overlaps between adjacent camera views are too small to be effective for training NeRFs.
Since the ego car is moving fast, some objects or contents only appear in a limited number of image views.
( e.g. Most of the vehicles need to be reconstructed from just 2 ∼views.)
All these problems make it difficult to optimize existing NeRFs for street-view synthesis.
MipNeRF-360 (Barron et al., 2022) is designed for training in unbounded scenes.
BlockNeRF (Tancik et al., 2022) proposes a block-combination strategy with refined poses, appearances, and exposure on the MipNeRF (Barron et al., 2021) base model for processing large-scale outdoor scenes.
However, they still require enough intersected camera rays (Figure 1(a)) and large overlaps across different cameras ( e.g. Block-NeRF uses a special system with twelve cameras for data acquisition to guarantee enough overlaps between different camera views).
They produce many blurs, “floaters” and other artifacts when training on existing self-driving datasets ( e.g. nuScenes (Caesar et al., 2019) and Waymo (Sun et al., 2020), as shown in Figure 2(a)).
Urban-NeRF (Rematas et al., 2022) takes accurate dense LiDAR depth as supervision for the reconstruction of urban scenes.
However, these dense LiDAR depths are difficult and expensive to collect.
As shown in Figure 3(a), data Caesar et al.
; Sun et al.
(2020) collected by self-driving cars
[Figure]
require can not be used for training Urban-NeRF because they only acquire sparse LiDAR points with plenty of outliers when projected to images ( e.g. only 2 ∼ 5K points are captured for each nuScenes image).
In this paper, we contribute a new NeRF design (named S-NeRF) for the novel view synthesis of both the large-scale (background) scenes and the foreground moving vehicles.
Different from other large-scale NeRFs (Tancik et al., 2022; Rematas et al., 2022), our method does not require specially designed data acquisition platform as used in Tancik et al.
; Rematas et al.
Our SNeRF can be trained on the standard self-driving datasets ( e.g. nuScenes (Caesar et al., 2019) and
Waymo (Sun et al., 2020)) that are collected by common self-driving cars with fewer cameras and noisy sparse LiDAR points to synthesize novel street views.
We improve the scene parameterization function and the camera poses for learning better neural representations from street views.
We also develop a novel depth rendering and supervision method using the noisy sparse LiDAR signals to effectively train our S-NeRF for street-view synthesis.
To deal with the depth outliers, we propose a new confidence metric learned from the robust geometry and reprojection consistencies.
Not only for the background scenes, we further extend our S-NeRF for high-quality reconstruction of the moving vehicles ( e.g. moving cars) using the proposed virtual camera transformation.
Note that this is impracticable for previous NeRFs (Mildenhall et al., 2020;
Deng et al., 2022; Barron et al., 2021; 2022).
In the experiments, we demonstrate the performance of our S-NeRF on the standard driving datasets (Caesar et al., 2019; Sun et al., 2020).
For the static scene reconstruction, our S-NeRF far outperforms the large-scale NeRFs (Barron et al., 2021; 2022; Rematas et al., 2022).
It reduces the mean-squared error by 7 ∼ 40% and produces impressive depth renderings (Figure 2(b)).
For the foreground objects, S-NeRF is shown capable of reconstructing moving vehicles in high quality, which is impracticable for conventional NeRFs (Mildenhall et al., 2020; Barron et al., 2021;
Deng et al., 2022).
It also beats the latest mesh-based reconstruction method Chen et al.
(2021), improving the PSNR by 45% and the structure similarity by 18%.
R ELATED W ORK
Reconstructing 3D geometries and rendering photorealistic novel views from multi-view images and their camera poses have a wide range of application scenarios.
2. 3D R ECONSTRUCTION
Traditional reconstruction and novel view rendering (Agarwal et al., 2011) rely on Structure-fromMotion (SfM), multi-view stereo and graphic rendering (Losasso & Hoppe, 2004).
Learning-based approaches have been widely used in 3D scene and object reconstruction (Sitzmann et al., 2019; Xu et al., 2019; Engelmann et al., 2021).
They encode the feature through a deep neural network and learn various geometry representations, such as voxels (Kar et al., 2017; Sitzmann et al., 2019), patches (Groueix et al., 2018) and meshes (Wang et al., 2018;
Chen et al., 2021).
2. N EURAL R ADIANCE F IELDS
Neural Radiance Fields (NeRF) is proposed in (Mildenhall et al., 2020) as an implicit neural representation for novel view synthesis.
Various types of NeRFs haven been proposed for acceleration.
[Figure]
(Yu et al., 2021a; Rebain et al., 2021;
Wizadwongsa et al., 2021), better generalization abilities (Yu et al., 2021b; Trevithick & Yang, 2021), new implicit modeling functions (Yariv et al., 2021; Wang et al., 2021a), large-scale scenes (Barron et al., 2021; Tancik et al., 2022; Zhang et al., 2020), and depth supervised training (Deng et al., 2022; Rematas et al., 2022).
Large-scale NeRF Many NeRFs have been proposed to address the challenges of large-scale outdoor scenes.
NeRF in the wild (Martin-Brualla et al., 2021) applies appearance and transient embeddings to solve the lighting changes and transient occlusions.
Neural plenoptic sampling (Li et al., 2021b) proposes to use a Multi-Layer Perceptron (MLP) as an approximator to learn the plenoptic function and represent the light-field in NeRF.
Mip-NeRF (Barron et al., 2021) develops a conical frustum encoding to better encode the scenes at a continuously-valued scale.
Using Mip-NeRF as a base block, Block-NeRF (Tancik et al., 2022) employs a block-combination strategy along with pose refinement, appearance, and exposure embedding on large-scale scenes.
Mip-NeRF 360 (Barron et al., 2022) improves the Mip-NeRF for unbounded scenes by contracting the whole space into a bounded area to get a more representative position encoding.
Depth supervised NeRF DS-NeRF (Deng et al., 2022) utilizes the sparse depth generated by
COLMAP (Sch¨onberger & Frahm, 2016) to supervise the NeRF training.
PointNeRF
(Xu et al., 2022) uses point clouds to boost the training and rendering with geometry constraints.
DoNeRF (Neff et al., 2021) realizes the ray sampling in a log scale and uses the depth priors to improve ray sampling.
NerfingMVS (Wei et al., 2021) also instructs the ray sampling during training via depth and confidence.
Dense depth priors are used in (Roessle et al., 2022) which are recovered from sparse depths.
These methods, however, are designed for processing small-scale objects or indoor scenes.
UrbanNeRF
(Rematas et al., 2022) uses accurate dense LiDAR depth as supervision to learn better reconstruction of the large-scale urban scenes.
But these dense LiDAR depths are difficult and expensive to collect.
The common data collected by self-driving cars (Caesar et al., 2019; Sun et al., 2020) can not be used in training Urban-NeRF because they only acquire noisy and sparse
LiDAR points.
In contrast, our S-NeRF can use such defect depths along with a learnable confidence measurement to learn better neural representations for the large-scale street-view synthesis.
N E RF FOR S TREET V IEWS
In this section, we present our S-NeRF that can synthesize photo-realistic novel-views for both the large-scale background scenes and the foreground moving vehicles (Section 3.3).
[Figure]
In the street views, there are many dynamic objects.
In order to be used for self-driving simulation or VR applications, dynamic objects must be fully controllable and move in controlled locations, speed, and trajectories.
Therefore, the background scenes and the foreground vehicles must be reconstructed separately and independently.
We propose a novel NeRF design that uses sparse noisy LiDAR signals to boost the robust reconstruction and novel street-view rendering.
Pose refinement and the virtual camera transform are added to achieve accurate camera poses (Section 3.2).
To deal with the outliers in the sparse LiDAR depths, we use a depth completion network Park et al.
(2020) to propagate the sparse depth and employ a novel confidence measurement based on the robust reprojection and geometry confidence (Section 3.4).
Finally, S-NeRF is trained using the proposed depth and RGB losses (Section 3.5).
3. P RELIMINARY
Neural Radiance Field (NeRF) represents a scene as a continuous radiance field and learns a mapping function f : ( x , θ ) → ( c , σ ) .
It takes the 3D position x i ∈ R 3 and the viewing direction θ i as input and outputs the corresponding color c i with its differential density σ i .The
mapping function is realized by two successive multi-layer perceptrons (MLPs).
NeRF uses the volume rendering to render image pixels.
For each 3D point in the space, its color can be rendered through the camera ray r ( t ) =
o + t d with N stratified sampled bins between the near and far bounds of the distance.
The output color is rendered as:
[Equation]
where o is the origin of the ray, T i is the accumulated transmittance along the ray, c i and σ i are the corresponding color and density at the sampled point t i .
δ j = t j +1 − t j refers to the distance between the adjacent point samples.
3. C AMERA P OSE P ROCESSING
SfM (Sch¨onberger & Frahm, 2016) used in previous NeRFs (Mildenhall et al., 2020; Deng et al., 2022; Barron et al., 2021; 2022) fails in computing camera poses for the self-driving data since the camera views have fewer overlaps (Figure 1(b)).
Therefore, we proposed two different methods to reconstruct the camera poses for the static background and the foreground moving vheicles.
Background scenes For the static background, we use the camera parameters achieved by sensorfusion SLAM and IMU of the self-driving cars (Caesar et al., 2019; Sun et al., 2020) and further reduce the inconsistency between multi-cameras with a learning-based pose refinement network.
We follow Wang et al.
Wang et al.
(2021b) to implicitly learn a pair of refinement offsets ∆ P = (∆ R, ∆ T ) for each original camera pose P =
[ R, T ] , where R ∈ SO (3) and T ∈ R 3 .
The pose refine network can help us to ameliorate the error introduced in the SLAM algorithm and make our system more robust.
Moving vehicles While the method proposed above is appropriate for the static background, camera pose estimation of moving objects is especially difficult due to the complicated movements of both the ego car and the target objects.
As illustrated in Figure 4, we compute the relative position ˆ
P between the camera and the target object.
We use the center of the target object as the origin of the coordinate system.
In the experiments, we use the 3D detectors ( e.g. Yin et al.
to detect the 3D bounding box and the center of the target object.
Using the ego car’s center as the coordinate system’s origin.
P b represents the position of the target object.
P i is the position of the camera
i (there are 5 ∼ 6 cameras on the ego car).
We now transform
[Figure]
the coordinate system by setting the target object’s center as the coordinate system’s origin (as illustrated in Figure 4).
[Equation]
Here, P = (cid:20) R
T 0 T (cid:21) represents the old position of the camera or the target object.
ˆ P
i is the new relative camera position.
3. R EPRESENTATION OF S TREET S CENES
In a self-driving system ( e.g. Geosim Chen et al.
(2021)), vehicles on the road play the most important role in driving decisions.
There are some important application scenarios ( e.g. expressway) only contain cars.
In this paper, we mainly focus on the background scenes and moving vehicles.
Other rigid and nonrigid objects can also use the similar techniques.
Background scenes The street view sequences in nuScenes and Waymo datasets usually span a long range ( > 200m), so it is necessary to constrain the whole scene into a bounded range before the position encoding.
The Normalized Device Coordinates (NDC) (Mildenhall et al., 2020) fails in our scenes due to the complicated motions of the ego car and the challenging camera settings (Figure 1(b)).
We improve the scene parameterization function used in Barron et al.
as:
[Equation]
Where r is the radius parameter to decide the mapping boundary.
The mapping is independent of the far bounds and avoids the heavy range compressing of the close objects, which gives more details in rendering the close objects (controlled by r ).
We use frustum from Mip-NeRF (Barron et al., 2021), sampling along rays evenly in a log scale to get more points close to the near plane.
Moving Vehicles
In order to train NeRF with a limited number of views ( e.g. 2 ∼ 6 image views), we compute the dense depth maps for the moving cars as an extra supervision.
We follow GeoSim (Chen et al., 2021) to reconstruct coarse mesh from multi-view images and the sparse
LiDAR points.
After that, a differentiable neural renderer (Liu et al., 2019) is used to render the corresponding depth map with the camera parameter (Section 3.2).
The backgrounds are masked during the training by an instance segmentation network (Wang et al., 2020).
3. D EPTH S UPERVISION
As illustrated in Figure 3, to provide credible depth supervisions from defect LiDAR depths, we first propagate the sparse depths and then construct a confidence map to address the depth outliers.
Our depth confidence is defined as a learnable combination of the reprojection confidence and the geometry confidence .
During the training, the confidence maps are jointly optimized with the color rendering for each input view.
LiDAR depth completion We use NLSPN (Park et al., 2020) to propagate the depth information from LiDAR points to surrounding pixels.
While NLSPN performs well with 64-channel LiDAR data ( e.g. KITTI Geiger et al.
(2012; 2013) dataset), it doesn’t generate good results with nuScenes’ 32-channel LiDAR data, which are too sparse for the depth completion network.
As a result, we accumulate neighbour LiDAR frames to get much denser depths for NLSPN.
These accumulated
LiDAR data, however, contain a great quantity of outliers due to the moving objects, ill poses and occlusions, which give wrong depth supervisions.
To address this problem, we design a robust confidence measurement that can be jointly optimized while training our S-NeRF.
Reprojection confidence To measure the accuracy of the depths and locate the outliers, we first use an warping operation ψ to reproject pixels X =
( x, y, d ) from the source images I s to the target image I t .
Let P s , P t be the source and the target camera parameters, d ∈ D s as the depth from the source view.
The warping operation can be represented as:
[Equation]
ψ represents the warping function that maps 3D points to the camera plane and ψ − 1 refers to the inverse operation from 2D to 3D points.
Since the warping process relies on depth maps D s , the depth outliers can be located by comparing the source image and the inverse warping one.
We introduce the RGB, SSIM (Wang et al., 2004) and the pre-trained VGG feature (Simonyan & Zisserman, 2015) similarities to measure the projection confidence in the pixel, patch structure, and feature levels:
[Equation]
Where ˆ I s =
I t ( X t ) is the warped RGB image, and the ˆ F s = F t ( X t ) refers to the feature reprojection.
The receptive fields of these confidence maps gradually expand from the pixels and the local patches to the non-local regions to construct robust confidence measurements.
Geometry confidence We further impose a geometry constrain to measure the geometry consistency of the depths and flows across different views.
Given a pixel X s =
( x s , y s , d s ) on the depth image
D s we project it to a set of target views using equation 4.
The coordinates of the projected pixel X t = ( x t , y t , d t ) are then used to measure the geometry consistency.
For the projected depth d t , we compute its consistency with the original target view’s depth ˆ d t = D t ( x t , y t ) :
[Equation]
For the flow consistency, we use the optical flow method (Zhang et al., 2021) to compute the pixel’s motions from the source image to the adjacent target views f s → t .
The flow consistency is then formulated as:
[Equation]
Where τ is a threshold in γ to identify the outliers through the depth and flow consistencies.
Learnable confidence combination To compute robust confidence map, we define the learnable weights ω for each individual confidence metric and jointly optimize them during the training.
The final confidence map can be learned as ˆ C = (cid:80) i ω i
C i , where (cid:80)
i ω i
=
The i ∈ { rgb, ssim, vgg, depth, flow } represents the optional confidence metrics.
The learnable weights ω adapt the model to automatically focus on correct confidence.
3. L OSS F UNCTIONS
Our loss function consists of a RGB loss that follows (Mildenhall et al., 2020; Barron et al., 2021;
Deng et al., 2022), and a confidence-conditioned depth loss.
To boost the depth learning, we also employ edge-aware smoothness constraints, as in (Li et al., 2021a; Godard et al., 2019), to penalize large variances in depth according to the image gradient | ∂I | :
[Equation] [Equation]
[Table]
[Figure]
Where R is a set of rays in the training set.
For the reconstruction of the foreground vehicles, D refers to the depth.
And in background scenes, D represents the disparity (inverse depth) to make the model focus on learning important close objects. λ 1 and λ 2 are two user-defined balance weights.
E XPERIMENTS 4. E XPERIMENTAL S ETUP
We perform our experiments on two open source self-driving datasets: nuScenes (Caesar et al., 2019) and Waymo (Sun et al., 2020).
We compare our S-NeRF with the state-of-the-art methods (Barron et al., 2021; 2022; Rematas et al., 2022; Chen et al., 2021).
For the foreground vehicles, we extract car crops from nuScenes and Waymo video sequences.
For the large-scale background scenes, we use scenes with 90 ∼ 180 images.
In each scene, the ego vehicle moves around 10 ∼ 40 meters, and the whole scenes span more than 200m. We do not test much longer scenes limited by the single
NeRF’s representation ability.
Our model can merge different sequences like Block-NeRF (Tancik et al., 2022) and achieve a larger city-level representation.
In all the experiments, the depth and smooth loss weight λ 1 and λ 2 are set to 1 and 0 .
15 respectively for foreground vehicles.
And for background street scenes, we set τ = 20% for confidence measurement and the radius r = 3 in all scenes.
λ 1 = 0 .
2 and λ 2 = 0 .
01 are used as the loss balance weights.
More training details are available in the supplementary materials.
4. N OVEL V IEW R ENDERING FOR F OREGROUND V EHICLES
In this section, we present our evaluation results for foreground vehicles.
We compare our method with the latest non-NeRF car reconstruction method (Chen et al., 2021).
Note that existing
NeRFs (Mildenhall et al., 2020; Deng et al., 2022) cannot be used to reconstruct the moving vehicles.
To compare with the NeRF baseline (Mildenhall et al., 2020), we also test our method on the static cars.
Since COLMAP (Sch¨onberger & Frahm, 2016) fails in reconstruct the camera parameters here, the same camera poses used by our S-NeRF are applied to the NeRF baseline to implement comparisons.
Static vehicles Figure 5 and Table 1 show quantitative and visualized comparisons between our method and others in novel view synthesis.
Optimizing NeRF baseline on a few (4 ∼ 7) image views leads to severe blurs and artifacts.
GeoSim produces texture holes when warping textures for novel view rendering.
The shapes of the cars are also broken due to the inaccurate meshes.
In contrast, our S-NeRF shows more fine texture details and accurate object shapes.
It can improve the PSNR and SSIM by 45 ∼ 65% compared with the NeRF and GeoSim baselines.
[Figure]
[Table]
Moving vehicles As compared in Figure 2(c) ∼ 2(d) and Table 1, novel view synthesis by
GeoSim (Chen et al., 2021) is sensitive to mesh errors that will make part of the texture missing or distorted during novel view warping and rendering.
In contrast, our S-NeRF provides larger ranges for novel view rendering than geosim (see supplementary) and generates better synthesis results.
S-NeRF can also simulate the lighting changing for different viewing directions, which is impossible for GeoSim.
Furthermore, S-NeRF does not heavily rely on accurate mesh priors to render photorealistic views.
The confidence-based design enables S-NeRF to eliminate the geometry inconsistency caused by depth outliers.
S-NeRF surpasses the latest mesh based method (Chen et al., 2021) by 45% in PSNR and 18% in SSIM.
4. N OVEL V IEW R ENDERING FOR B ACKGROUND S CENES
Here we demonstrate the performance of our S-NeRF by comparing with the state-of-the-art methods Mip-NeRF (Barron et al., 2021), Urban-NeRF (Rematas et al., 2022) and Mip-NeRF 360 (Barron et al., 2022).
We use the offical code of Mip-Nerf for evaluation and expand the hidden units of the MLP to 1024 (the same as ours).
Since there is no official code published, we tried our best to reproduce (Rematas et al., 2022; Barron et al., 2022) based on our common settings.
We test four nuScenes sequences and report the evaluation results in Table 2.
Our method outperforms
Mip-NeRF, Mip-NeRF 360 and Urban NeRF in all three evaluation metrics.
We see significant improvements of 40% in PSNR and 26% in SSIM and a 45% reduction in LPIPS compared with the
Mip-NeRF baseline.
It also outperforms the current best Mip-NeRF 360 by 7.5% in PSNR, 4.5% in
SSIM and 5% in LPIPS.
We also show 360-degree panoramic rendering in Figure 6.
S-NeRF significantly ameliorates artifacts, suppresses “floaters” and presents more fine details compared with
Urban-NeRF and Mip-NeRF 360.
Results of Waymo scenes are shown in the appendix.
[Table]
[Figure]
4. B ACKGROUND AND F OREGROUND F USION
There are two different routes to realize controllable foreground and background fusion: depthguided placement and inpainting ( e g. GeoSim Chen et al.
and joint NeRF rendering ( e g.
GIRAFFE Niemeyer & Geiger (2021)).
Both of them heavily rely on accurate depth maps and 3D geometry information for object placement, occlusion handling, etc. .
Our method can predict far better depth maps and 3D geometry ( e.g. meshes of cars) than existing NeRFs (Zhang et al., 2020;
Barron et al., 2022).
We provide a video (in the supplementary materials) to show the controlled placement of vehicles to the background using the depth-guided placement.
4. A BLATION S TUDY
To further demonstrate the effectiveness of our method, we conduct a series of ablation studies with or without certain component.
Table 3 shows the quantitative results of our model ablations.
We evaluate our model on four foreground vehicles and two background scenes under different settings.
Visualized results are also provided in Figure 7.
More ablation studies are available in the appendix.
For background scene rendering, our RGB baseline outperforms the Mip-NeRF by 56% in meansquared error and 42% in structure similarity.
Using inaccurate depth supervision without confidence leads to a drop of the accuracy due to the depth outliers.
Confidence contributes to about 3% improvement in PSNR and SSIM.
PSNR of our model slightly drops after adding the edge-aware smooth loss to it.
However, it effectively suppresses “floaters” and outliers to improve our depth quality.
For moving vehicles, the depth supervision and confidence measurement improves the RGB baseline by 18% and 8% in PSNR individually.
The smoothness loss mainly improves the structure similarity and reduces the LPIPS errors.
C ONCLUSION , L IMITATIONS AND F UTURE W ORK
We contribute a novel NeRF design for novel view rendering of both the large-scale scenes and foreground moving vehicles using the steet view datasets collected by self-driving cars.
In the experiments, we demonstrate that our S-NeRF far outperforms the state-of-the-art NeRFs with higherquality RGB and impressive depth renderings.
Though S-NeRF significantly outperforms MipNeRF and other prior work, it still produces some artifacts in the depth rendering.
For example, it produces depth errors in some reflective windows.
In the future, we will use the block merging as proposed in Block-NeRF (Tancik et al., 2022) to learn a larger city-level neural representation.
B I MPLEMENTATION D ETAILS B. D EPTH COMPLETION
We use NLSPN[Park et al.
network for depth completion, which propagates the depth information from sparse LiDAR points to surrounding pixels.
While NLSPN performs well with 64-channel LiDAR data, such as KITTIGeiger et al.
(2012; 2013) and waymoSun et al.
Ettinger et al.
(2021) datasets, it doesn’t generate good results with nuScenesCaesar et al.
(2019) 32-channel LiDAR data, which are too sparse to complete the depth.
We accumulate 5 ∼ 10 neighbour LiDAR frames to get a much denser LiDAR data for the NLSPN network.
These accumulated
LiDAR points, however, contains many outliers due to the moving objects, ill poses, occlusions and reprojection errors.
These outliers will give wrong depth information.
To remove these outliers, we first compute the optical flow Zhang et al.
(2021) of one RGB image using its neighbour frame.
After that, we reproject each LiDAR points to neighbour image plane to get the LiDAR flow, and then compare two kinds of flows to locate the LiDAR outliers using a threshold of 20% (following Eq.
(6 ∼ 8) of the paper).
Applying above procedure, we can remove many outliers.
However, some of these outliers still exists due to the errors of the optical flow and ill poses.
These outliers still exist after depth completion.
The depth completion algorithm also introduces new outliers to the final dense depth map, which are great challenges for depth supervision.
We therefore learn an confidence metric for more robust depth supervision.
B. R EMOVE M OVING O BJECTS IN S TREET V IEWS
Currently, we focus on static scenes.
When training the background scenes, we masked the moving objects, while static objects ( e.g. static vehicles) are kept and trained along with the background.
The moving vehicles are trained independently.
Other moving objects ( e.g. person) can be removed by instance segmentation and optical flow.
B. F OREGROUND V EHICLES
For foreground vehicles, we use four layers with 256 hidden units in the MLP.
The depth and smooth loss weights λ 1 and λ 2 are set to 1 and 0 .
15 respectively.
We sample 64 times along each ray during the RGB and depth rendering.
We train our S-NeRF for 30k iterations using Adam optimizer with 5 − 4 as the learning rate and 1024 as the batch size.
The training takes about 2 hours for each vehicle on a single RTX3090 gpu.
For each vehicle, there are around 2 ∼ 8 views used for training and 1 ∼views for testing.
B. S TREET V IEWS (B ACKGROUND )
For background street scenes, we set τ = 20% for the confidence measurement and the radius r =for the scene parameterization function.
Our coarse-fine network share the same parameters like mipNeRF.
The density MLP has eight layers with 1024 hidden units and the color MLP consists of three layers with 128 hidden units.
We keep this setting for all evaluation methods for fair comparisons.
We train our S-NeRF for 100 K iterations using Adam optimizer with a batch size of 2048.
The learning rate is reduced log-linearly from 5 × 10 − 4 to 5 × 10 − 6 with a warm-up phase of 2500 iterations.
λ 1 = 0 .
2 and λ 2 = 0 .
01 are used as the loss balance weights.
We sample 128 times along each ray in a log scale.
Our S-NeRF is trained on two RTX3090 GPUs which takes about 17 hours for a scene with about 250 images (with a resolution of 1280 × 1920).
C E XPERIMENTS
We present more visualized results in Figure 7 and Figure 9.
We also provide a video demo for performance illustration.
[Table] [Table]
C. P ARAMETERS AND E FFICIENCY
In the experimennts [Table] we use the same settings for the MLP encoding in our S-NeRF and other state-of-the-art methods Barron et al.
(2021; 2022); Rematas et al.
There are 8.76M learnable parameters in our S-NeRF that is similar to other state-of-the-art methods Barron et al.
Rematas et al.
(2022) (8.7 ∼ 9.9M).
All the methods are trained for 100k iterations.
Our method takes about 17 hours in training for one street scene.
This is the same as Mip-NeRF and Urban-NeRF because we use the same settings during the experiments.
Mip-NeRF 360 is faster than ours because it doesn’t require the coarse rendering for supervision.
This strategy can also be used in our S-NeRF to accelerate the convergence and further improve the quality of the novel view rendering.
C. W AYMO R ESULTS
We also test our S-NeRF on two Waymo street-view sequences.
The results are reported in Tableand visually compared with the state-of-the-arts Rematas et al.
; Barron et al.
(2022) in Figure 8 and 9.
Waymo dataset use a 64-channel LiDAR and five cameras for capture driving scenes.
Our S-NeRF outperforms Mip-NeRF by 40% in PSNR, 80% in SSIM and 44% in LPIPS.
It’s also far better than the Urban-NeRF baseline (32 ∼ 50 % ↑ in PSNR, SSIM and LPIPS) which also uses the sparse LiDAR depth as supervision.
Compared with the current best Mip-NeRF 360, our method also achieves a 6.8% improvements in PSNR and a 5.2% reduction of the LPIPS error.
More importantly, our S-NeRF predicts much more accurate depths for the large-scale street views.
We also use Waymo dataset to compare our method with the NeRF baseline on the vehicle synthesis.
The quantitive evaluations are shown in Table 2 and visually compared in Figure 7.
Our method outperforms the NeRF baseline by 63% in PSNR, 18% in SSIM and 51% in LPIPS.
Compared with the recently proposed mesh-based car reconstruction method Chen et al.
(2021b), our S-NeRF improves the PSNR by 62% and reduces the LPIPS error by 16%.
C. C OMPARISONS WITH U RBAN -N E RF
Urban-NeRF also use the LiDAR depth to boost the NeRF training.
However, it requires accurate
LiDAR depth.
Most of the street view datasets ( e.g. Waymo and nuScenes) contains plenty of depth
[Table] [Table]
outliers due to the influence of ill poses, reprojection errors, occlusion and moving objects.
We find that these depth outliers can heavily influence the rendering quality of Urban-NeRF.
Our SNeRF introduces confidence metrics to make it more robust to the depth outliers.
Moreover, our
S-NeRF improves the scene parameterization function and the camera poses to learn better neural representation for rendering large-scale street views.
As shown in Figure 9 and 8, compared with
Urban-NeRF, our S-NeRF gives more details in street-view objects (such as the car, the tree and road lines) and are more robust for the large-scale street views captured by the self-driving cars.
C. M ORE A BLATION STUDY
In this section, we use two nuScenes street-view scenes and three foreground vehicles to test the effects of different settings when training our S-NeRF.
These include different loss balance weights, different confidence components, different depth qualities and different scene parameterizations.
In these ablation experiments, S-NeRF is trained for 30k iterations.
Loss balance weights As shown in Table 3 and 4, we compare the performance of our S-NeRF using different loss balance weights.
Our S-NeRF is not very sensitive to the changes of the loss balance weights.
Using larger or smaller λ 1 or λ 2 just slightly reduces the PSNR and SSIM by 0.2 ∼ 1% on the background novel view rendering.
4] (for λ 1 ) and (0 .
02) (for λ 2 ) are the reasonable range for our loss balance weights for training street background views.
For the foreground vehicles, [0 . 5 , 2] (for λ 1 ) and (0 . 075 , 0 . 015) (for λ 2 ) are the reasonable range.
Confidence components We also test the performances of our S-NeRF when using different confidence components (geometry and reprojection measurements) for learning our confidence metric.
As shown in Table 5, when we remove the reprojection confidence module, the PSNR slightly dropped by 0.4%.
And when we train S-NeRF without geometry confidence, the PSNR and SSIM is about 0.7% lower.
We also test the effects of the threshold τ used in the geometry confidence (Eq.
(7) of the paper).
We find that the geometry confidence is not sensitive to the threshold τ .
[10% , 40%] is a reasonable range for the threshold τ .
For the foreground vehicles, we only use RGB and SSIM confidences.
This is because the depth map used in training vehicles are relatively better than the backgrounds.
Thus, we do not need strong geometry confidences.
We report these ablation results in Table 6 by using only RGB confidence
Reporjection
Geometry
[Table] [Table]
or SSIM confidence.
We find that using only RGB or SSIM confidence could achieve a little better
PSNR (1 ∼ 2% ↑ ), but a relative worse LPLPS ( 6 . 6 ∼ 11 % ↓ ).
Taken all these three evaluation metrics into consideration, using both RGB and SSIM confidence gives a better performance in training our S-NeRF.
We also report depth error rate in Table 7.
For the accurate depths, it predicts high confidence to encourage the NeRF geometry to be consistent with the LiDAR signal.
Depth quality As shown in Table 9, we study the effects of different depth map qualities.
We train our S-NeRF using depth map in different qualities.
To simulate the depth errors in different qualities, we add random Gaussian Noise to the original depth map inputs.
The strength of the noise (the quality of the depths) is measured by PSNR and error rates compared with the original depth inputs.
Error rate means how many outliers are introduced by the noise.
We use threshold =to compute the outlier rates compared with the original depth inputs.
Our method achieve similar rendering qualities when training with light noises, which means the light noises doesn’t influence the performance of our S-NeRF.
When training with strong noises, the PSNR and SSIM just slightly dropped by 0.05 ∼ 0.5%.
This shows that our method is robust to depth noises because our confidence strategy can locate and measure the depth outliers accurately and avoid the negative influence of the depth noises in training our S-NeRF.
Besides, we also test the performance of our S-NeRF when using only sparse depth for supervision (Table 8).
It performs better than our RGB-only S-NeRF (improving the PSNR by 5%).
But, it is 2% worse in PSNR and 8% worse in LPIPS than our default settings where dense depth map and the proposed confidence metric are used.
This means that dense depth maps can provide more useful geometry information for training our S-NeRF even though they may contain more depth outliers.
In addition, we also test another two worse depth completion methods by replacing the NLSPN with Ku et al.
Van Gansbeke et al.
(2019)(Table 10).
Even using the worse traditional depth completion method Ku et al.
(2018)(without learning a deep neural network), our S-NeRF still achieves a similar rendering quality (24.41 → 24.40).
This experiment shows that our method doesn’t rely on NLSPN depth quality.
Benefiting from our confidence-guided depth supervision, many other depth completion methods can also be used in our S-NeRF.
Scene and ray parameterization We studied the effects of using different radius in the scene parameterization function (Eq.
(2) of the paper).
Radius r is used to constrain the whole large-scale scene into a bounded range.
Close and far points are parameterized by different distance mapping function to make our S-NeRF able to keep more details for the close objects.
This is controlled by the radius parameter.
In Table 11, we report the performance of our S-NeRF when using different radius for scene parameterization.
We find that choosing r in 3 ∼ 10m could produce a better results than the
[Table] [Table]
original setting in Mip-Nerf 360Barron et al.
This is because the street views in nuScenes and Waymo datasets usually span a long range ( > 200m).
While, Mip-Nerf 360Barron et al.
(2022) do not have r to adjust the scene parameterization as ours.
Influence of quality of 3D detection results As shown in Table 12, the quality of 3D object detector does not significantly influence the rendering quality.
This is because we use pose refinement to improve the inaccurate initial results of the virtual camera pose estimation.
The pose refinement is guided by depth supervision and visual multiview constraints in training our S-NeRF.
We test the rendering results by using 3D bounding boxes of different qualities (mIoU: 0.79, 0.67 and 0.55) and compare them with the ground truth bounding boxes (achieved from the dataset labels).
We find that the PSNR doesn’t drop significantly compared with the ground truth bounding boxes.
When the mIoU of the bounding boxes is 0.79 (using the default detector), the PSNR just slightly drops by 0.23, the SSIM drops by 0.04, and the LPIPS doesn’t change.
When we use poor detection results, our S-NeRF still produces good results with a slight drop in the rendering quality.
(E.g. 0.49 ↓ when the mIoU is 0.67 and 0.97 ↓ for a poor mIoU of 0.55).
We observed that the detected bounding boxes ( > 0.5 in IoU) are good enough for training our S-NeRF.
This can be easily realized by many existing 3D object detectors.
Different pose refinements We have tried different pose refinements like NeRF–, BARF and SCNeRF.
As shown in Table 14, We find that NeRF– helps achieve the best quality when training on the self-driving dataset.
Possibly because NeRF– is more straightforward.
It directly learns shifts to the translation and rotation.
The shifts are relatively easier to learn under depth supervision.
It’s also possible that such shifts are easier to estimate when there is a small overlapped field of view between different cameras (as illustrated in Figure 1 of the paper.)
Training and Inference time Referring to Table 13, we show more details of our method against the existing large-scale NeRFs with the training time and inference time.
Our method (default settings) uses the same training setting and network parameters as those of MipNeRF.
They take the same time in training and inference.
When we implement the MipNeRF-360 strategy (the distillation
[Table] [Table]
mode, all other settings are kept the same), the light proposal MLP reduces both the training time and the inference time.
It runs 30% faster in training and 40% faster in inference than the default settings (with a drop of 0.33 in PSNR).
D L IMITATIONS AND S OCIAL I MPACTS
Failure cases.
Considering that the ego vehicle sometimes goes very fast, fewer images are captured by the side (left or right) cameras.
Some objects/contents only appeared in one or two left/right images.
This makes it hard to render high-quality left and right novel views.
As shown in Table 15, since there are fewer views, the rendered side (left/right) views report worse PSNR, SSIM and LPIPS ( 5 ∼ 12% ↓ ) compared with the front and back views.
In the future, we will try some techniques (e g. using a better method to densify depth or provide semantic supervision) to further improve the quality of rendering left and right views.
Potential risks.
Since our method can be used to render realistic street views and vehicles, it’s possible that S-NeRF can be used to synthesize fake photos or videos.
We therefore hope our S-NeRF could be used with cautiousness.
Work that bases itself on our method should also carefully consider the consequences of this potential negative social influence