基于半监督学习的煤矿井下行人检测模型_中国煤炭行业知识服务平台

基于半监督学习的煤矿井下行人检测模型

作者

陈湘源饶天荣潘涛
单位

国能榆林能源有限责任公司国能数智科技有限公司智能矿山与智慧运输事业部上海微系统与信息技术研究所无线传感网与通信重点实验室
摘要

在井工煤矿辅助运输车辆无人驾驶及安全监测领域，行人检测技术至关重要。目前已有许多工作进行了相应研究，但这些研究均需要使用大量的精确标注图片，而煤矿井下图片获取较为困难，标注难度也十分大，这些因素都极大地影响了相关模型的落地应用以及推广。为了解决这一问题，提出了一种基于半监督学习的煤矿井下行人检测模型。通过使用半监督学习框架可以有效降低煤矿井下行人检测模型对于高质量标注数据的大量需求。此外，针对煤矿井下设备运算能力较低的特点，还对YOLOv5模型进行了改进，在维持模型检测精度的条件下提升了模型的检测速度。实验表明基于半监督学习的煤矿井下行人检测模型可以使用仅相当于原数据集5%的数据训练得到较为有效的检测模型，大幅度较少了对于标注数据的依赖，对煤矿井下行人检测模型的快速应用和推广起到了帮助作用。
关键词

井下行人检测半监督学习YOLOv5
文章目录

0前言
1 半监督学习介绍
2 改进YOLOv5模型介绍
3 实验与分析
3.1 数据集与评价指标
3.2 对比和分析
4 结语