Pyhton实现点云保存为bin文件

本文介绍了一种LiDAR数据处理方法,包括从文本文件中读取LiDAR数据,并将其转换为二进制格式进行存储的过程。此方法适用于批量处理LiDAR数据集,提高数据加载效率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

# -*- coding: utf-8 -*-
"""
Created on Mon Nov 21 2022
@author: MMG
"""
# coding=utf-8

def data2bin(points, no):
    processed_dir = '/home/processed_lidar'
    points.astype(np.float32).tofile(os.path.join(processed_dir, str(no)))

def main():
     root_dir = '/home/gmm/snowyweather/Lidar'  # txt数据集路径
    cat = os.listdir(root_dir)
    iteration_num = len(cat)
    for i in range(iteration_num):
        filename = os.path.join(root_dir, cat[i])
        db_np = read_velodyne_txt(filename)        # 这个函数需要自己写
        data2bin(fiterdata, cat[i])


if __name__ == main():
    main()

### 将ROS 2中的点云消息(PointCloud2)保存为.bin文件 为了实现这一目标,可以编写一个简单的ROS 2节点来订阅`/point_cloud_topic`主题上的`sensor_msgs/msg/PointCloud2`消息,并将其转换成二进制格式的数据再写入`.bin`文件。下面是一个Python脚本的例子: #### Python代码示例 ```python import rclpy from sensor_msgs.msg import PointCloud2 import numpy as np import struct import os def pointcloud2_to_bin(msg, file_path): """ Convert a ROS PointCloud2 message to binary format and save it. Args: msg (PointCloud2): The input point cloud data received via the subscriber. file_path (str): Path where .bin files will be saved. Returns: None """ pc_data = list(point.fields for point in ros_point_cloud2.read_points(msg)) points = [] for p in pc_data: x = float(p[0]) y = float(p[1]) z = float(p[2]) # Append only valid points with finite coordinates if not np.isinf(x) and not np.isnan(x) \ and not np.isinf(y) and not np.isnan(y)\ and not np.isinf(z) and not np.isnan(z): points.append([x, y, z]) points_np = np.array(points).astype(np.float32) # Ensure directory exists before saving dir_name = os.path.dirname(file_path) if not os.path.exists(dir_name): os.makedirs(dir_name) # Save as bin file using little endian encoding points_np.tofile(file_path) class PCDtoBinNode(rclpy.node.Node): def __init__(self): super().__init__('pcd_to_bin_node') self.subscription_ = self.create_subscription( PointCloud2, '/point_cloud_topic', self.listener_callback, 10) self.file_index = 0 def listener_callback(self, msg): filename = f"/path/to/save/location/frame_{self.file_index}.bin" pointcloud2_to_bin(msg, filename) self.get_logger().info(f'Saved {filename}') self.file_index += 1 if __name__ == '__main__': rclpy.init() node = PCDtoBinNode() try: rclpy.spin(node) except KeyboardInterrupt: pass node.destroy_node() rclpy.shutdown() ``` 此程序会监听指定的主题并将接收到的每一条点云信息存储在一个新的`.bin`文件中。每次当有新数据到来时都会增加索引号以区分不同的帧。 需要注意的是,在实际应用之前可能需要调整路径以及确保安装了必要的库如`ros2-py`和`numpy`等[^1]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值