Skip to content

Add BDD100K detection dataset loader#554

Open
JaisalJain wants to merge 1 commit into
JdeRobot:masterfrom
JaisalJain:feature/bdd100k-detection-dataset
Open

Add BDD100K detection dataset loader#554
JaisalJain wants to merge 1 commit into
JdeRobot:masterfrom
JaisalJain:feature/bdd100k-detection-dataset

Conversation

@JaisalJain
Copy link
Copy Markdown
Contributor

@JaisalJain JaisalJain commented Apr 18, 2026

Closes #515

Adds a BDD100K detection dataset loader, following the same pattern as the existing YOLO loader.

Files changed

  • perceptionmetrics/datasets/bdd100k.py: BDD100KDetectionDataset class and build_bdd100k_dataset() function
  • perceptionmetrics/datasets/__init__.py: registers BDD100K in the REGISTRY
  • tests/test_bdd100k.py: 5 mocked tests, no dataset download needed to run them
  • examples/bdd100k_detection.py: shows how to load the dataset

Expected directory layout

<images_dir>/train/*.jpg
<images_dir>/val/*.jpg
<labels_dir>/train/*.json  (one JSON per image)
<labels_dir>/val/*.json

Tested on real data

Loaded 80,000 samples (70k train, 10k val). Ground truth boxes shown below.
b1c66a42-6f7d68ca
b1c81faa-3df17267
b1c81faa-c80764c5

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.

Add BDD100K detection dataset loader

1 participant