Skip to content

Commit 574c8aa

Browse files
committed
add 3d bbox ref frame
1 parent 21f293b commit 574c8aa

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

GEMstack/onboard/perception/cone_detection_parking.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -354,7 +354,7 @@ def viz_object_states(self, cv_image, boxes):
354354
# Create bbox markers from cone dimensions
355355
ros_delete_bboxes_markers = delete_bbox_marker()
356356
self.pub_cones_bboxes_markers.publish(ros_delete_bboxes_markers)
357-
ros_cones_bboxes_markers = create_bbox_marker(cone_3d_centers, cone_3d_dims)
357+
ros_cones_bboxes_markers = create_bbox_marker(cone_3d_centers, cone_3d_dims, "lidar_top")
358358
self.pub_cones_bboxes_markers.publish(ros_cones_bboxes_markers)
359359

360360

GEMstack/onboard/perception/visualization_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def create_point_cloud(points, color=(255, 0, 0)):
6464

6565
return pc2.create_cloud(header, fields, point_cloud_data)
6666

67-
def create_bbox_marker(centroids, dimensions):
67+
def create_bbox_marker(centroids, dimensions, ref_frame="map"):
6868
"""
6969
Create 3D bbox markers from centroids and dimensions
7070
"""
@@ -76,7 +76,7 @@ def create_bbox_marker(centroids, dimensions):
7676
continue
7777

7878
marker = Marker()
79-
marker.header.frame_id = "top_lidar" # Reference frame
79+
marker.header.frame_id = ref_frame # Reference frame
8080
marker.header.stamp = rospy.Time.now()
8181
marker.ns = "bounding_boxes"
8282
marker.id = i # Unique ID for each marker

0 commit comments

Comments
 (0)