Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

SelfOcc: Self-Supervised Vision-Based 3D Occupancy Prediction

About

3D occupancy prediction is an important task for the robustness of vision-centric autonomous driving, which aims to predict whether each point is occupied in the surrounding 3D space. Existing methods usually require 3D occupancy labels to produce meaningful results. However, it is very laborious to annotate the occupancy status of each voxel. In this paper, we propose SelfOcc to explore a self-supervised way to learn 3D occupancy using only video sequences. We first transform the images into the 3D space (e.g., bird's eye view) to obtain 3D representation of the scene. We directly impose constraints on the 3D representations by treating them as signed distance fields. We can then render 2D images of previous and future frames as self-supervision signals to learn the 3D representations. We propose an MVS-embedded strategy to directly optimize the SDF-induced weights with multiple depth proposals. Our SelfOcc outperforms the previous best method SceneRF by 58.7% using a single frame as input on SemanticKITTI and is the first self-supervised work that produces reasonable 3D occupancy for surround cameras on nuScenes. SelfOcc produces high-quality depth and achieves state-of-the-art results on novel depth synthesis, monocular depth estimation, and surround-view depth estimation on the SemanticKITTI, KITTI-2015, and nuScenes, respectively. Code: https://github.com/huang-yh/SelfOcc.

Yuanhui Huang, Wenzhao Zheng, Borui Zhang, Jie Zhou, Jiwen Lu• 2023

Related benchmarks

TaskDatasetResultRank
3D Occupancy PredictionOcc3D-nuScenes (val)
mIoU9.3
213
3D Semantic Occupancy PredictionOcc3D
RayIoU9.6
40
3D Semantic Occupancy PredictionOcc3D-nuScenes v1.0 (val)
mIoU44.1
26
RGB ReconstructionnuScenes (val)
PSNR20.67
21
Depth EstimationnuScenes (val)
RMSE6.706
18
Occupancy EstimationOpenOcc v2
RayIoU9.1
12
Novel depth synthesisnuScenes
RMSE8.46
10
3D Occupancy PredictionSemanticKITTI v1.0 (test)
IoU21.97
9
Semantic Occupancy EstimationOcc3D-nuScenes
mIoU10.5
9
Novel depth synthesisSemanticKITTI
Abs Rel0.1562
7
Showing 10 of 22 rows

Other info

Code

Follow for update