lecrop-data
收藏资源简介:
TerraSentia冠层下导航数据集是一个专注于农业机器人领域的数据集,包含了TerraSentia机器人在作物冠层下进行导航任务时收集的多元时序数据。该数据集源自ROS1数据包,并经过处理以适用于机器学习任务。数据集包含三种配置:cpp配置使用作物跟随/纯追踪控制器,记录路径信息;crow配置使用iLQR控制器,记录作物行、目标点和iLQR时间信息;lecrop配置使用MPPI/强化学习结合视觉的方法,记录距离误差、航向误差、MPPI调试信息和关键点数据。数据以episodes形式组织,每个episode代表机器人连续运动片段,包含多列时间序列数据,每列对应一个ROS主题信号,并配有独立的时间戳向量。此外,每个episode还包含MP4/H.264格式的视频数据,如RGB相机视频、LiDAR点云可视化图和关键点可视化视频。数据字段包括元数据、机器人状态信号、控制命令信号、任务相关信号以及特定于lecrop配置的感知与控制信号。数据集在转换过程中移除了ROS数据包头信息和重型原始传感器数据,以精简结构,适用于机器人导航、控制策略学习、农业自动化、时序预测、模仿学习、强化学习及多模态机器学习模型的研究与开发。
The TerraSentia Canopy Navigation Dataset is a dataset focused on the agricultural robotics domain, containing multivariate time-series data collected during navigation tasks of the TerraSentia robot under crop canopies. It originates from ROS1 data bags and has been processed for machine learning tasks. The dataset includes three configurations: the cpp configuration uses a crop-following/pure pursuit controller and records path information; the crow configuration uses an iLQR controller and records crop rows, target points, and iLQR timing information; the lecrop configuration uses an MPPI/reinforcement learning approach combined with vision, recording distance errors, heading errors, MPPI debugging information, and keypoint data. Data is organized into episodes, each representing a continuous motion segment of the robot, with multiple columns of time-series data corresponding to ROS topic signals and independent timestamp vectors. Additionally, each episode includes MP4/H.264 format video data, such as RGB camera videos, LiDAR point cloud visualizations, and keypoint visualization videos. Data fields encompass metadata, robot state signals, control command signals, task-related signals, and perception and control signals specific to the lecrop configuration. During conversion, ROS data bag headers and heavy raw sensor data were removed to streamline the structure, making it suitable for research and development in robot navigation, control strategy learning, agricultural automation, time-series prediction, imitation learning, reinforcement learning, and multimodal (vision + state) machine learning models.
数据集概要
- 数据集名称: TerraSentia Under-Canopy Navigation
- 任务类型: 机器人学(robotics)
- 标签: 机器人学、农业、冠层下、导航、rosbag、terrasentia
- 许可证: 其他(other)
- 主要语言: 葡萄牙语(描述性文字)
- 来源: 基于 ROS1 bags 提取的 TerraSentia 农业机器人在作物冠层下(under-canopy)的导航数据。
数据配置
数据集包含三个子配置,对应不同的控制器来源:
| 配置名 | 规模(episodes) | 来源 / 控制器描述 | 特殊话题特征 |
|---|---|---|---|
cpp |
143 | Crop-follow / pure-pursuit 控制器 | 记录 path 话题 |
crow |
31 | iLQR 控制器 | 记录 crop_lines, goal, ilqr_time |
lecrop |
137 | MPPI/RL + 视觉控制器 | 记录 dist_err, head_err, mppi_*, keypoint |
数据结构
- 每行 = 一个 episode: 代表机器人在地里行驶直至停下的一段连续轨迹。通过里程计速度(经平滑处理)进行分割:行驶速度 > 0.05 m/s 判定为“移动”;静止时间 >= 5 秒则开启新 episode。时长不足 2 秒或消息数少于 20 的 episode 被丢弃。无里程计的 bag 被作为一个完整 episode。
- 每列 = 一个信号(ROS topic): 存储为列表形式的时间序列。不同话题具有不同采样率,每个话题拥有独立的时间向量
*_t(单位秒,相对于 episode 开始)。 - 视频列:
rgb、lidar_plot、keypoint_vis_*采用Video类型(MP4/H.264 编码),每个 episode 一个视频,帧率约 10 fps,时间信息记录在对应的*_t列中。 - 已移除字段: 原始 ROS 的元数据(
header,seq,stamp,frame_id,covariance,layout)、原始深度传感器和 LiDAR/PointCloud 点云数据、以及tf、camera_info等中间传递话题。
数据列说明
元数据(标量)
source,episode,bag,duration_s,n_msgs
信号列(列表类型,每组附有时间向量 <g>_t)
| 组名 | 包含列 | 原始 ROS 话题 |
|---|---|---|
odom_* |
odom_t, odom_pos_{x,y,z}, odom_quat_{x,y,z,w}, odom_vel_{x,y,z}, odom_angvel_{x,y,z} |
/…/dlio/odom_node/odom |
imu_* |
imu_t, imu_acc_{x,y,z}, imu_gyro_{x,y,z}, imu_quat_{x,y,z,w} |
/…/imu |
cmd_* |
cmd_t, cmd_lin_{x,y,z}, cmd_ang_{x,y,z} |
/…/cmd_vel |
motion_* |
motion_t, motion_lin_{x,y,z}, motion_ang_{x,y,z} |
/…/motion_command |
path_* |
path_t, path_pos_{x,y,z}, path_quat_{x,y,z,w}(多折线,每个时间步一个列表) |
/…/path |
goal_* |
goal_t, goal_pos_{x,y,z}, goal_quat_{x,y,z,w} |
/…/goal(仅 crow) |
crop_lines_* |
crop_lines_t, crop_lines_{m1,b1,m2,b2}(表示作物行的直线参数) |
/…/crop_lines(仅 crow) |
ilqr_* |
ilqr_t, ilqr_time |
/…/ilqr_time(仅 crow) |
dist_err, head_err |
横向和航向误差(预测值) | /…/*_error_predicted(仅 lecrop) |
mppi_dist, elite_scores, value_info |
MPPI/RL 调试信息,每个为列表的列表 | /…/rl_debug/*(仅 lecrop) |
keypoint |
视觉关键点,列表的列表 | /…/vision/keypoint(仅 lecrop) |
视频列(Video 类型,MP4/H.264,约 10 fps,帧时间记录在对应 *_t 列)
| 列名 | 内容 | 原始 ROS 话题 |
|---|---|---|
rgb |
RGB 摄像头视频(640×360) | /…/rgb/image_rect_color/compressed(所有配置) |
lidar_plot |
LiDAR 点云与作物行绘图视频(224×224) | /lidar_plot(仅 crow) |
keypoint_vis_argmax, keypoint_vis_heatmap |
关键点可视化视频 | /…/vision/keypoint_vis_*/compressed(仅 lecrop) |
注意: 特定配置独有的列在其他配置的 episode 中显示为空(null)。例如
cpp配置中的部分cropfollowpp_lecropfollowbags 也会携带lecrop的列。
数据来源与预处理
- 原始数据: 来自
cpp_runs、crow_runs、lecrop_runs三个 bag 文件夹(包括 rebuttal 经历和已命名的 runs,相同名称的重复 bag 已被去重)。 - 生成流程:
rosbags→ 基于移动状态进行运动段分割 → 转换为 Parquet 格式。
使用示例(Python)
python from datasets import load_dataset ds = load_dataset("arthurpompeu/lecrop-data", "cpp", split="train") ep = ds[0]
ep["odom_pos_x"], ep["odom_vel_x"], ep["cmd_lin_x"], ...




