Skip to content

Commit cdc9f63

Browse files
author
Tianyu Li
committed
[bugfix] fix can_bus format in [PR](in OpenDriveLab#214)
1 parent 1e49d55 commit cdc9f63

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

projects/mmdet3d_plugin/datasets/nuscenes_e2e_dataset.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -554,7 +554,8 @@ def get_data_info(self, index):
554554
translation = input_dict['ego2global_translation']
555555
can_bus = input_dict['can_bus']
556556
can_bus[:3] = translation
557-
can_bus[3:7] = rotation
557+
# NOTE(lty): fix can_bus format, in https://github.com/OpenDriveLab/UniAD/pull/214
558+
can_bus[3:7] = rotation.elements
558559
patch_angle = quaternion_yaw(rotation) / np.pi * 180
559560
if patch_angle < 0:
560561
patch_angle += 360

0 commit comments

Comments
 (0)