关键词:
反光衣检测
YOLO
聚类
SPPC(空间金字塔约束)
嵌入式系统
摘要:
近年来,在建筑、交通和工业等诸多行业中,由于施工人员未穿戴反光衣而发生伤亡事故的现象层出不穷。当前大多数行业对施工人员穿戴反光衣的监管过度依赖人工,这种人工筛查工人是否穿戴反光衣的方法不仅效率低下,而且由于角度、距离、光线等复杂场景的变化,很容易对工人是否穿着反光衣做出误判。基于此,本文借助YOLOV4-tiny和YOLOV7-tiny算法开展对反光衣穿戴智能检测方法的研究,以提高反光衣穿戴检测的效率和精度,并将其部署在嵌入式平台,为后续大规模工业应用提供可行性和可靠性的参考。主要研究内容如下:
(1)针对YOLOV4-tiny算法的检测精度与检测速度的耦合问题,本文提出了一种YOLO-FL算法以兼顾原YOLOV4-tiny算法的检测精度与检测速度。首先,提出一种新的交并比计算方法KIoU,并将其引入到原K-means++聚类方法中,避免原K-means++聚类方法中距离带来的影响;其次,引入SPPF模块到颈部网络中,融合不同尺度的特征信息以获得更丰富的特征表示;最后,在模型的训练过程中引入标签平滑策略,降低模型因数据样本较少而发生过拟合的概率。在反光衣数据集上进行实验,YOLO-FL算法的平均精度mAP和F1值分别为91.14%和0.85,比原算法分别提高了 8.29%和0.12,FPS为116,表明YOLO-FL算法能在提高检测精度的同时,兼顾检测速度的实时性要求。
(2)针对YOLOV7-tiny算法特征提取能力不足与特征融合不充分等原因容易导致的误检和漏检问题,本文提出了一种YOLO-JS算法。首先,引入ODConv卷积模块替代YOLOV7-tiny算法颈部网络中的部分传统卷积,使核空间的四个维度都被赋予卷积动态属性以提高基本卷积运算的特征提取能力,从而提高其信息处理和分析的效率;其次,引入PConv卷积模块替代YOLOV7-tiny颈部网络中的另一部分传统卷积,使之同时缩减冗余计算和内存访问;最后,提出一种新的SPPC空间金字塔池化模块,替代原算法网络结构中的SPPCSPC模块,旨在不增加计算开销的情况下,充分融合上下文信息。通过在反光衣数据集上的实验结果表明,YOLO-JS算法的平均精度mAP和F1值均优于其他主流算法,分别为96.10%和0.93,比原算法分别提高了 6.60%和0.06,GFLOPs为12.40。YOLO-JS算法不仅提高了反光衣穿戴检测的精度,还进一步改善了原算法存在的漏检和误检问题,且满足检测的实时性要求。
(3)为了满足实际工业应用的需要,本文将YOLO-JS算法部署到嵌入式设备Jetson Nano。通过TensorRT推理框架缩减模型计算精度与重构网络结构的方式提升模型在嵌入式设备的推理速度。实验观测表明,YOLO-JS算法在嵌入式设备上依旧保持较高的检测精度与实时检测效率,为模型在未来展开大规模工业应用部署打下坚实的基础。