矿业安全与环保2024,Vol.51Issue(4) :35-40.DOI:10.19835/j.issn.1008-4495.20240325

基于HigherHRNet的煤矿井下人体姿态估计快速网络研究

Research on fast network of human pose estimation in underground coal mine based on HigherHRNet

张延军 陈博
矿业安全与环保2024,Vol.51Issue(4) :35-40.DOI:10.19835/j.issn.1008-4495.20240325

基于HigherHRNet的煤矿井下人体姿态估计快速网络研究

Research on fast network of human pose estimation in underground coal mine based on HigherHRNet

张延军 1陈博1
扫码查看

作者信息

  • 1. 太原科技大学 机械工程学院,山西 太原 030024
  • 折叠

摘要

煤矿井下人体姿态的快速估计是井下作业智慧安全检测的重要前提.为解决煤矿井下多尘多雾、照明不足及颜色相融等问题,提高人体姿态估计关键点分配准确度及网络运行速度,研究新的Optimising HigherHRNet(OH-HRNet)快速网络模型:对HigherHRNet模型的轻量化设计、关键点分配进行深入研究,提出了基于注意力机制的记忆卷积模块及强化骨骼约束的关键点分配算法,并改进了算法的损失函数.在煤矿井下场景数据集和COCO公开数据集上的实验结果表明:OH-HRNet在GPU的速度上是 LitePose 的 1.06 倍,平均精度均值 mAP 提高了 7.4%,平均召回率均值 mAR 提高了14.0%,可以实现更有效的智慧安全检测.

Abstract

Fast estimation of human pose in underground coal mine is an important prerequisite for intelligent safety detection of underground operation.Aiming at the problems of dusty and foggy,insufficient illumination and color blending in underground coal mine,this study conducts an in-depth study on the lightweight design and key point assignment of the HigherHRNet model and proposes a new Optimising HigherHRNet(OH-HRNet)fast network model in order to improve the accuracy of the key point assignment for human pose estimation as well as the network operation speed.The OH-HRNet model proposes a memory convolution module based on attention mechanism and a key point assignment algorithm with reinforced skeletal constraints,and the loss function of the algorithm is improved.Experiments on the coal mine underground scenario dataset and COCO public dataset show that,OH-HRNet is 1.06 times faster than LitePose in terms of GPU speed,with a 7.4%increase in mAP and a 14.0%increase in mAR,which can achieve more effective intelligent safety detection.

关键词

智慧煤矿/人体姿态估计/快速网络/关键点检测/关键点分配

Key words

intelligent coal mine/human pose estimation/fast network/key point detection/key point assignment

引用本文复制引用

基金项目

山西省重点研发项目(202102010101010)

出版年

2024
矿业安全与环保
中煤科工集团重庆研究院,国家煤矿安全技术工程研究中心

矿业安全与环保

CSTPCD北大核心
影响因子:0.987
ISSN:1008-4495
段落导航相关论文