摘要
本发明提供了一种矿井工人违规目标检测方法、系统、设备及存储介质,属于图像识别与处理领域。针对矿井下目标尺度变化大的问题,基于原始YOLO8模型在主干网络加入多层EMA注意力机制,增强模型对尺度变化适应能力和特征提取能力。其次,引入统一自注意力机制检测头DyHead替换原有检测头,提升检测头特征表达能力。在自建数据集和公开数据集VOC2012上的平均精度均有所提高,泛化能力很强。与其他模型相比,在具有更低模型复杂度的同时具有更高检测精度,更适合矿井下复杂环境的目标检测。
技术关键词
注意力机制
检测头
识别矿井
图片
作业场景
特征提取能力
复杂度
网络
数据获取模块
语义
通道
处理器
精度
计算机设备
代表
动态