一种基于ST-GCN的临边洞口作业不安全行为检测方法及装置

本发明提供一种基于ST‑GCN的临边洞口作业不安全行为检测方法及装置，涉及视觉识别技术领域。所述基于ST‑GCN的临边洞口作业不安全行为检测方法包括：获取施工人员临边洞口作业动作的已标注视频数据集；采用改进的YOLOv5算法以及OpenPose算法对已标注视频数据集进行处理，获得临边洞口作业动作的骨架节点数据集；构建初始的ST‑GCN神经网络模型；根据骨架节点数据集，对初始的ST‑GCN神经网络进行训练，获得训练好的ST‑GCN神经网络模型；获取待识别的施工人员临边洞口作业动作的视频数据；将待识别的施工人员临边洞口作业动作的视频数据，输入训练好的ST‑GCN神经网络模型中进行识别，获得施工人员临边洞口作业不安全行为的识别结果。采用本发明可提高施工现场安全管理人员的工作效率。