MCPcopy Index your code
hub / github.com/OpenDriveLab/OccNet

github.com/OpenDriveLab/OccNet @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
248 symbols 908 edges 118 files 98 documented · 40%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

Scene as Occupancy

We believe Occupancy serves as a general representation of the scene and could facilitate perception and planning in the full-stack of autonomous driving.

https://github.com/OpenDriveLab/OccNet/assets/54334254/92fb43a0-0ee8-4eab-aa53-0984506f0ec3

Scene as Occupancy - Paper in arXiv | CVPR 2023 AD Challenge Occupancy Track - Point of contact: simachonghao@pjlab.org.cn

3D Occupancy Prediction Leaderboard

We provide a full-scale 3D occupancy leaderboard based on the CVPR 2023 Autonomous Driving challenge. Top entries (by 06-09-2023) are provided below. Check the website out!

teaser

Table of Contents

Highlights

teaser - :oncoming_automobile: General Representation in Perception: 3D Occupancy is a geometry-aware representation of the scene. Compared to the form of 3D bounding box & BEV segmentation, 3D occupancy could capture the fine-grained details of critical obstacles in the scene. - :trophy: Exploration in full-stack Autonomous Driving: OccNet, as a strong descriptor of the scene, could facilitate subsequent tasks such as perception and planning, achieving results on par with LiDAR-based methods (41.08 on mIOU in 3D occupancy, 60.46 on mIOU in LiDAR segmentation, 0.703 avg.Col in motion planning).

News

Getting Started

Results and Pre-trained Models

We will release pre-trained weight soon.

teaser

TODO List

  • [x] 3D Occupancy and flow dataset v1.0
  • [x] 3D Occupancy Prediction code v1.0
  • [ ] Pre-trained Models
  • [ ] Occupancy label generation code
  • [ ] GT label with more voxel size
  • [ ] Compatibility with other BEV encoders

License & Citation

All assets (including figures) and code are under the Apache 2.0 license unless specified otherwise. The data license inherits the license used in nuScenes dataset.

Please consider citing our paper if the project helps your research with the following BibTex:

@article{sima2023_occnet,
      title={Scene as Occupancy}, 
      author={Chonghao Sima and Wenwen Tong and Tai Wang and Li Chen and Silei Wu and Hanming Deng  and Yi Gu and Lewei Lu and Ping Luo and Dahua Lin and Hongyang Li},
      year={2023},
      eprint={2306.02851},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Challenge

We host the first 3D occupancy prediciton challenge on CVPR 2023 End-to-end Autonomous Driving Workshop. For more information about the challenge, please refer to here.

Related resources

Awesome - BEVFormer - BEV Perception Survey & Recipe

Core symbols most depended-on inside this repo

trans_matrix
called by 3
tools/ray_iou/ego_pose_extractor.py
init_weight
called by 3
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py
extract_feat
called by 3
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py
trans_matrix
called by 3
projects/mmdet3d_plugin/datasets/ego_pose_extractor.py
build_dataloader
called by 3
projects/mmdet3d_plugin/datasets/builder.py
process_one_sample
called by 3
projects/mmdet3d_plugin/datasets/ray_metrics.py
process_one_sample
called by 2
tools/ray_iou/ray_casting.py
get_scene_token
called by 2
tools/ray_iou/ego_pose_extractor.py

Shape

Method 147
Function 56
Class 44
Route 1

Languages

Python98%
C++2%

Modules by API surface

projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py21 symbols
projects/mmdet3d_plugin/models/backbones/vovnet.py20 symbols
projects/mmdet3d_plugin/bevformer/detectors/bevformer_occ.py12 symbols
projects/mmdet3d_plugin/bevformer/modules/spatial_cross_attention.py9 symbols
projects/mmdet3d_plugin/bevformer/modules/decoder.py9 symbols
tools/ray_iou/ego_pose_extractor.py8 symbols
projects/mmdet3d_plugin/models/utils/grid_mask.py8 symbols
projects/mmdet3d_plugin/datasets/ego_pose_extractor.py8 symbols
projects/mmdet3d_plugin/bevformer/modules/encoder.py8 symbols
utils/vis.py7 symbols
tools/ray_iou/ray_casting.py7 symbols
projects/mmdet3d_plugin/core/evaluation/eval_hooks.py7 symbols

For agents

$ claude mcp add OccNet \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact