计算机与现代化 ›› 2025, Vol. 0 ›› Issue (10): 1-6.doi: 10.3969/j.issn.1006-2475.2025.10.001
• 图像处理 • 下一篇
摘要:
摘要:针对厂区人员异常行为识别中背景复杂、计算成本受限等问题,本文提出一种基于YOLOv5的轻量级工厂人员异常行为识别网络。该网络结合全维动态卷积模块(Omni-dimensional Dynamic Convolution, ODConv)和显式视觉中心模块(Explicit Visual Center Block, EVCBlock),以更低的参数计算量实现更好的检测表现。在颈部网络中引入的ODConv模块增强模型对复杂工厂环境的适应性,并减少模型参数量。在骨干网络末端增加的EVCBlock模块主要目的是提高模型的检测精度,以弥补参数降低给模型带来过量的精度损失。此外,利用归一化Wasserstein距离(Normalized Wasserstein Distance, NWD)构建改进的损失函数,优化模型的训练过程并提高对小目标的检测效果。为了验证模型性能,本文利用现有的轻量化方法额外构建多种改进的识别模型。对比实验的结果表明,与现有方法相比本文提出的轻量级识别模型具有更好的检测精度,同时模型参数比原始模型更少。相比原始模型,本文构建检测模型的mAP提升了3.2个百分点,GFLOPs下降了2.2。该模型对实现工业生产场景中工厂人员异常行为的快速检测和准确识别有重要意义。
中图分类号: