Improving 3D Occupancy Prediction through Class-balancing Loss and Multi-scale Representation
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866929358156857344 |
|---|---|
| author | Chen, Huizhou Wang, Jiangyi Li, Yuxin Zhao, Na Cheng, Jun Yang, Xulei |
| author_facet | Chen, Huizhou Wang, Jiangyi Li, Yuxin Zhao, Na Cheng, Jun Yang, Xulei |
| contents | 3D environment recognition is essential for autonomous driving systems, as autonomous vehicles require a comprehensive understanding of surrounding scenes. Recently, the predominant approach to define this real-life problem is through 3D occupancy prediction. It attempts to predict the occupancy states and semantic labels for all voxels in 3D space, which enhances the perception capability. Birds-Eye-View(BEV)-based perception has achieved the SOTA performance for this task. Nonetheless, this architecture fails to represent various scales of BEV features. In this paper, inspired by the success of UNet in semantic segmentation tasks, we introduce a novel UNet-like Multi-scale Occupancy Head module to relieve this issue. Furthermore, we propose the class-balancing loss to compensate for rare classes in the dataset. The experimental results on nuScenes 3D occupancy challenge dataset show the superiority of our proposed approach over baseline and SOTA methods. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_16099 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Improving 3D Occupancy Prediction through Class-balancing Loss and Multi-scale Representation Chen, Huizhou Wang, Jiangyi Li, Yuxin Zhao, Na Cheng, Jun Yang, Xulei Computer Vision and Pattern Recognition 3D environment recognition is essential for autonomous driving systems, as autonomous vehicles require a comprehensive understanding of surrounding scenes. Recently, the predominant approach to define this real-life problem is through 3D occupancy prediction. It attempts to predict the occupancy states and semantic labels for all voxels in 3D space, which enhances the perception capability. Birds-Eye-View(BEV)-based perception has achieved the SOTA performance for this task. Nonetheless, this architecture fails to represent various scales of BEV features. In this paper, inspired by the success of UNet in semantic segmentation tasks, we introduce a novel UNet-like Multi-scale Occupancy Head module to relieve this issue. Furthermore, we propose the class-balancing loss to compensate for rare classes in the dataset. The experimental results on nuScenes 3D occupancy challenge dataset show the superiority of our proposed approach over baseline and SOTA methods. |
| title | Improving 3D Occupancy Prediction through Class-balancing Loss and Multi-scale Representation |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2405.16099 |