fix roi_boxes noise translation

open-mmlab · Jul 31, 2023 · b4dd915 · b4dd915
1 parent 8a76f59
commit b4dd915
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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