fix roi_boxes noise translation

Author: David Josef Emmerichs

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