基于深度学习和模型压缩技术的轻量级煤矿人车检测模型 -——以贵州地区煤矿为例
解北京1
李恒1
栾铮1
雷振2
李晓旭1
李卓3
1.中国矿业大学(北京)应急管理与安全工程学院,北京 1000832.贵州理工学院矿业工程学院,贵州贵阳 5500033.北京工业大学城市建设学部,北京 100124
摘要：煤矿工人和载人车辆(煤矿人车)的智能识别是视频监控系统的重要组成部分,也是煤矿智能化发展的关键任务.然而,煤矿人车检测场景较为复杂,大型人车检测模型部署在有限的计算设备上难以实现,如何在模型检测性能和检测效率之间取得平衡存在诸多挑战.以贵州地区煤矿视频监控数据集为例,提出了一种基于深度学习和模型压缩技术的轻量级煤矿人车检测模型,该模型精准实时的完成了煤矿人车检测任务,对网络进行瘦身的同时几乎没有损失检测性能.具体来说,在网络模型设计阶段,以YOLOv8s为基线提出了一种名为FCW-YOLO的煤矿人车轻量级检测模型,首先将Faster-Block和坐标注意力和开发到网络的特征提取模块中,设计了一种新颖的C2f-Faster-CA轻量级架构,旨在减少网络的冗余通道同时自适应捕捉全局关键信息;其次,采用了WIOU边界回归损失函数以增加模型对普通质量样本的关注,降低了训练样本不平衡带来的回归误差等问题.在模型压缩阶段,联动剪枝算法对提出的FCW-YOLO模型进行通道级别的稀疏,模型可自动识别不重要的通道并对其进行删减,实现了煤矿人车检测模型二次轻量化设计FCWP-YOLO.在自建的煤矿人车检测数据集上的结果表明,提出的模型参数量,计算量和模型大小分别为2.3 M,4.0GFLOPs,6.0 MB,对比基线模型分别实现了4.9倍、4.7倍、4.4倍的压缩效果,平均检测精度为88.7％,提高了1.1％,每张图像的处理时间仅为5.6 ms.对比多种轻量级架构和先进的检测模型,该方法精度表现优异,计算成本更低,实时性能更好,为资源受限的煤矿场景提供了一种可行的煤矿人车检测方法,满足煤矿视频监控部署要求,可为煤矿人车智能巡检任务提供实时预警.
关键词：煤矿工人检测煤矿载人车辆检测深度学习模型压缩轻量级架构
分类号：TD82(矿山开采)TP18(自动化基础理论)TP391(计算技术、计算机技术)
资助基金：国家重点研发计划(2022YFC2904100)国家重点研发计划(2023YFC3011300)中央高校基本科研业务费专项资金资助项目(2023ZKPYAQ04)
论文发表日期：2025-02-28
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：16( 1393-1408 )
英文信息