用图注意力+Transformer预测工地人员与障碍物轨迹,精度显著提升。
TrajGATFormer: A Graph-Based Transformer Approach for Worker and Obstacle Trajectory Prediction in Off-site Construction Environments
- 结合YOLOv10n与DeepSORT实现精准检测跟踪,用图注意力机制捕捉交互关系。
- 预测4.8秒后轨迹,人员误差仅1.25米(ADE),障碍物预测更准至1.15米。
- 适合智能安全系统研发者,尤其关注复杂工地动态避障场景。
随着建筑行业对高效、安全施工的需求增长,场外施工成为解决方案,但工人、机械与移动障碍物的紧密互动带来了新的安全风险。准确预测人员轨迹并考虑社会与环境因素,是构建防碰撞系统的关键。传统方法难以适应施工环境的动态性,常依赖简化假设或手工特征,限制了对复杂实时交互的建模能力。尽管数据驱动方法改善了时序模式捕捉,仍难处理长期行为及空间社交上下文。本文提出融合YOLOv10n与DeepSORT的检测跟踪框架,并设计两种新型轨迹预测模型:TrajGATFormer和TrajGATFormer-Obstacle。YOLOv10n负责在多样场景中精确识别工人与障碍物,DeepSORT实现持续追踪并赋予唯一ID。两模型均采用带图注意力网络(GAT)的Transformer编码器-解码器结构,以捕捉时空交互。在4.8秒预测时长下,TrajGATFormer对工人轨迹的平均位移误差(ADE)为1.25米,最终位移误差(FDE)为2.3米;而TrajGATFormer-Obstacle同时预测工人与障碍物,精度更高,达ADE 1.15米、FDE 2.2米。对比实验表明,两模型相较传统方法,分别降低35%与38%的误差。
原文摘要 · Abstract (English)
As the demand grows within the construction industry for processes that are not only faster but also safer and more efficient, offsite construction has emerged as a solution, though it brings new safety risks due to the close interaction between workers, machinery, and moving obstacles. Predicting the future trajectories of workers and taking into account social and environmental factors is a crucial step for developing collision-avoidance systems to mitigate such risks. Traditional methods often struggle to adapt to the dynamic and unpredictable nature of construction environments. Many rely on simplified assumptions or require hand-crafted features, limiting their ability to respond to complex, real-time interactions between workers and moving obstacles. While recent data-driven methods have improved the modeling of temporal patterns, they still face challenges in capturing long-term behavior and accounting for the spatial and social context crucial to collision risk assessment. To address these limitations, this paper proposes a framework integrating YOLOv10n and DeepSORT for precise detection and tracking, along with two novel trajectory prediction models: TrajGATFormer and TrajGATFormer-Obstacle. YOLOv10n serves as the backbone for object detection, accurately identifying workers and obstacles in diverse scenes, while DeepSORT efficiently tracks them over time with unique IDs for continuity. Both models employ a transformer encoder-decoder with Graph Attention Networks (GAT) to capture temporal and spatial interactions. TrajGATFormer predicts worker trajectories with an ADE of 1.25 m and FDE of 2.3 m over a 4.8 s horizon, while TrajGATFormer-Obstacle extends prediction to both workers and obstacles, achieving higher accuracy (ADE 1.15 m, FDE 2.2 m). Comparative analysis shows both models outperform traditional methods, reducing ADE and FDE by up to 35% and 38%, respectively.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。