HiveCoreRD/hivecore_robot_vision
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix problem

Browse Source
This commit is contained in:
liangyuxuan
2025-12-01 16:10:13 +08:00
parent e2884eecd1
commit e9afb039b0
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tools/get_image.py
View File

@@ -19,8 +19,12 @@ def main():
rclpy.init()
node = Node("get_image")
sub = node.create_subscription(ImgMsg, "/img_msg", sub_callback, 10)
node.get_logger().info("Waiting for image...")
while sign:
rclpy.spin_once(node)
node.get_logger().info("Image received")
node.destroy_node()
rclpy.shutdown()
def sub_callback(msg):
@@ -35,9 +39,9 @@ def sub_callback(msg):
cv2.imwrite("test/color_image.png", cv_bridge.imgmsg_to_cv2(camera_data[0]))
cv2.imwrite("test/depth_image.png", cv_bridge.imgmsg_to_cv2(camera_data[1]))
with open("test/K.json", "w") as f:
f.write(json.dumps(camera_data[2]))
f.write(json.dumps(camera_data[2].tolist()))
with open("test/D.json", "w") as f:
f.write(json.dumps(camera_data[3]))
f.write(json.dumps(camera_data[3].tolist()))
sign = False