From b4dd91578b3645ae503d55acda1ff691f913134b Mon Sep 17 00:00:00 2001 From: David Josef Emmerichs Date: Fri, 23 Jun 2023 17:56:13 +0200 Subject: [PATCH] fix roi_boxes noise translation --- pcdet/datasets/augmentor/data_augmentor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pcdet/datasets/augmentor/data_augmentor.py b/pcdet/datasets/augmentor/data_augmentor.py index 0945f1de6..fd0fef648 100644 --- a/pcdet/datasets/augmentor/data_augmentor.py +++ b/pcdet/datasets/augmentor/data_augmentor.py @@ -152,7 +152,7 @@ def random_world_translation(self, data_dict=None, config=None): gt_boxes[:, :3] += noise_translate if 'roi_boxes' in data_dict.keys(): - data_dict['roi_boxes'][:, :3] += noise_translate + data_dict['roi_boxes'][:, :, :3] += noise_translate data_dict['gt_boxes'] = gt_boxes data_dict['points'] = points