Skip to content

fix(pai-converter): center ego-vehicle bbox z-gate at height/2#163

Open
Willyzw wants to merge 1 commit into
NVIDIA:mainfrom
Willyzw:wei/fix-pai-ego-bbox-z
Open

fix(pai-converter): center ego-vehicle bbox z-gate at height/2#163
Willyzw wants to merge 1 commit into
NVIDIA:mainfrom
Willyzw:wei/fix-pai-ego-bbox-z

Conversation

@Willyzw

@Willyzw Willyzw commented Jul 16, 2026

Copy link
Copy Markdown

Description

Fixes a bug in the PAI converter's ego-vehicle bbox filter that leaves Lidar self-return points near the ego roof unfiltered.

Root cause. In tools/data_converter/pai/utils.py, the function filter_ego_vehicle_points only offsets the box center in x (rear_axle_to_bbox_center), while y and z stay centered at 0. According to the PAI coordinate-system docs, the rig z origin is at ground level, so the z-gate spans [-(h/2+pad), h/2+pad] instead of [ground-pad, roof+pad]. Thus the self-return points at upper roof part fall outside the box and are not filtered.

Fix. Center the z-gate at height/2 so the box spans from ground to roof:

Verification. Confirmed on a PAI clip as shown below, red = box before this fix, green = after (rear y-z views); the red box top cuts through the middle of ego vehicle while the green box encloses the vehicle entirely.

ego bbox before/after

Type of Change

  • Bug fix (non-breaking change that fixes an issue)
  • New feature (non-breaking change that adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to change)
  • Documentation update
  • Refactoring (no functional changes)

filter_ego_vehicle_points only offset the box center in x
(rear_axle_to_bbox_center); y and z stayed centered at 0. Since the rig z
origin is at ground level, the z-gate spanned [-(h/2+pad), h/2+pad]
instead of [ground-pad, roof+pad], leaving the upper body / roof
(z > height/2 + padding) outside the box. Ego self-returns near the
roof-mounted lidar were therefore not filtered and showed up as noise
within ~1.5 m of the sensor at roof height.

Center the z-gate at height/2 so the box spans ground..roof. Verified on
a PAI clip: near-sensor ego roof returns drop to ~0.
@copy-pr-bot

copy-pr-bot Bot commented Jul 16, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@janickm janickm self-assigned this Jul 20, 2026
@janickm
janickm self-requested a review July 20, 2026 07:23
@janickm janickm added the bug Something isn't working label Jul 20, 2026
@janickm

janickm commented Jul 21, 2026

Copy link
Copy Markdown
Collaborator

Verified fix locally, thanks for spotting and correcting @Willyzw !

Merging once CI is happy

@janickm

janickm commented Jul 21, 2026

Copy link
Copy Markdown
Collaborator

/ok to test be0876b

@janickm
janickm enabled auto-merge July 21, 2026 07:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants