feat: add header msg

mjlee111 · Nov 4, 2024 · 3eb4b95 · 3eb4b95
1 parent 5a79960
commit 3eb4b95
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/image_recognition/yolo_detection/yolo_detection/yolo_detection_node.py b/image_recognition/yolo_detection/yolo_detection/yolo_detection_node.py
@@ -146,6 +146,8 @@ def listener_callback(self, msg):
 
         for i, (cls, bbox) in enumerate(zip(results[0].boxes.cls, results[0].boxes.xyxy)):
             bounding_box_msg = BoundingBoxMsgs()
+            bounding_box_msg.header.frame_id = msg.header.frame_id
+            bounding_box_msg.header.stamp = self.get_clock().now().to_msg()
             bounding_box_msg.box = [float(bbox[0]), float(bbox[1]), float(bbox[2]), float(bbox[3])]
 
             class_idx = int(cls)