Skip to content

GitLab

Unverified Commit d0bb1847 authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

fix dbsampler bug (#1812)

parent 8751c079
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
mmdet3d/datasets/transforms/transforms_3d.py
View file @ d0bb1847
...@@ -367,11 +367,10 @@ class ObjectSample(BaseTransform): ...@@ -367,11 +367,10 @@ class ObjectSample(BaseTransform):
gt_bboxes_3d = input_dict['gt_bboxes_3d'] gt_bboxes_3d = input_dict['gt_bboxes_3d']
gt_labels_3d = input_dict['gt_labels_3d'] gt_labels_3d = input_dict['gt_labels_3d']
if self.use_ground_plane and 'plane' in input_dict['ann_info']: if self.use_ground_plane:
ground_plane = input_dict['plane'] ground_plane = input_dict.get('plane', None)
assert ground_plane is not None, '`use_ground_plane` is True ' \ assert ground_plane is not None, '`use_ground_plane` is True ' \
'but find plane is None' 'but find plane is None'
input_dict['plane'] = ground_plane
else: else:
ground_plane = None ground_plane = None
# change to float for blending operation # change to float for blending operation
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment