Skip to content

Add nuScenes Dataset Support for 2D Object Detection#397

Open
tejasstanley wants to merge 6 commits intoJdeRobot:masterfrom
tejasstanley:nuScenes_detection
Open

Add nuScenes Dataset Support for 2D Object Detection#397
tejasstanley wants to merge 6 commits intoJdeRobot:masterfrom
tejasstanley:nuScenes_detection

Conversation

@tejasstanley
Copy link

PR: Add nuScenes Dataset Support for 2D Object Detection

Summary

This PR adds support for the nuScenes dataset for 2D object detection by introducing the NuScenesDetectionDataset class. It parses 3D bounding boxes from nuScenes into 2D camera views, keeps canonical class labels, and works with the existing tutorials (tutorial_image_detection.ipynb).

Changes Included

  • New dataset class: NuScenesDetectionDataset inheriting from ImageDetectionDataset.
  • Added dependency on nuscenes-devkit in pyproject.toml.
  • Notebooks/examples updated for nuScenes integration.
  • Code formatted with Black, type annotations, and Sphinx-style docstrings included.

Notes / Pending Tasks

  • Dataset viewer tab integration for nuScenes is in progress and will be added in a follow-up PR.
  • Occluded projected boxes are kept as-is for now.
  • Working on adding nuScenes LiDAR support and a LiDAR segmentation tutorial; will update once ready.

Testing

  • Verified dataset creation, reading annotations, and tutorial execution on nuScenes v1.0-mini.
  • Visual inspection of bounding boxes works correctly with current pipeline.

References

@dpascualhe
Copy link
Collaborator

Hi @tejasstanley . Thanks for your contribution! It will take me a couple of days to review it, but at first sight, the docstrings don't follow the Sphinx style (reference: https://sphinx-rtd-tutorial.readthedocs.io/en/latest/docstrings.html). Can you update that? Also, can you share links so that I can download the dataset myself?

@tejasstanley
Copy link
Author

tejasstanley commented Mar 4, 2026

Hi @dpascualhe, thanks for the feedback, I will update the docstrings.
nuScenes can be downloaded by logging in this website https://www.nuscenes.org/nuscenes#download, and selecting the US version of the mini dataset. Direct link can't be copied it seems.
image

@dpascualhe dpascualhe self-requested a review March 8, 2026 18:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants