HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Cylindrical and Asymmetrical 3D Convolution Networks for LiDAR Segmentation

Xinge Zhu; Hui Zhou; Tai Wang; Fangzhou Hong; Yuexin Ma; Wei Li; Hongsheng Li; Dahua Lin

Cylindrical and Asymmetrical 3D Convolution Networks for LiDAR Segmentation

Abstract

State-of-the-art methods for large-scale driving-scene LiDAR segmentation often project the point clouds to 2D space and then process them via 2D convolution. Although this corporation shows the competitiveness in the point cloud, it inevitably alters and abandons the 3D topology and geometric relations. A natural remedy is to utilize the3D voxelization and 3D convolution network. However, we found that in the outdoor point cloud, the improvement obtained in this way is quite limited. An important reason is the property of the outdoor point cloud, namely sparsity and varying density. Motivated by this investigation, we propose a new framework for the outdoor LiDAR segmentation, where cylindrical partition and asymmetrical 3D convolution networks are designed to explore the 3D geometric pat-tern while maintaining these inherent properties. Moreover, a point-wise refinement module is introduced to alleviate the interference of lossy voxel-based label encoding. We evaluate the proposed model on two large-scale datasets, i.e., SemanticKITTI and nuScenes. Our method achieves the 1st place in the leaderboard of SemanticKITTI and outperforms existing methods on nuScenes with a noticeable margin, about 4%. Furthermore, the proposed 3D framework also generalizes well to LiDAR panoptic segmentation and LiDAR 3D detection.

Code Repositories

xinge008/Cylinder3D
Official
pytorch
Mentioned in GitHub
L-Reichardt/Cylinder3D-updated-CUDA
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-semantic-segmentation-on-scribblekittiCylinder3D
mIoU: 57.0
3d-semantic-segmentation-on-semantickittiCylinder3D
test mIoU: 68.9%
val mIoU: 64.3%
lidar-semantic-segmentation-on-nuscenesCylinder3D+InstanceAug
test mIoU: 0.77
robust-3d-semantic-segmentation-onCylinder3D (torchsparse)
mean Corruption Error (mCE): 103.13%
robust-3d-semantic-segmentation-onCylinder3D (spconv)
mean Corruption Error (mCE): 103.25%
robust-3d-semantic-segmentation-on-nuscenes-cCylinder3D (torchsparse)
mean Corruption Error (mCE): 105.56%
robust-3d-semantic-segmentation-on-nuscenes-cCylinder3D (spconv)
mean Corruption Error (mCE): 111.84%
robust-3d-semantic-segmentation-on-wod-cCylinder3D (torchsparse)
mean Corruption Error (mCE): 106.02%
semi-supervised-semantic-segmentation-on-23Sup.-only (Voxel)
mIoU (1% Labels): 39.2
mIoU (10% Labels): 48.0
mIoU (20% Labels): 52.1
mIoU (50% Labels): 53.8
semi-supervised-semantic-segmentation-on-25Sup.-only (Voxel)
mIoU (1% Labels): 50.9
mIoU (10% Labels): 65.9
mIoU (20% Labels): 66.6
mIoU (50% Labels): 71.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Cylindrical and Asymmetrical 3D Convolution Networks for LiDAR Segmentation | Papers | HyperAI