计算机与现代化 ›› 2024, Vol. 0 ›› Issue (08): 43-48.doi: 10.3969/j.issn.1006-2475.2024.08.008

• 人工智能 • 上一篇    下一篇

面向智慧运维系统的改进YOLOv8行为检测算法






  

  1. (国网安徽省电力有限公司信息通信分公司,安徽 合肥 230061)
  • 出版日期:2024-08-28 发布日期:2024-08-28
  • 基金资助:
    安徽省高等学校自然科学基金重点项目(KJ2020A0250)

Improved YOLOv8 Behavior Detection Algorithm for Intelligent Operation and#br# Maintenance System

  1. (Information and Telecommunication Branch, State Grid Anhui Electric Power Co., Ltd., Hefei 230061, China)
  • Online:2024-08-28 Published:2024-08-28

摘要: 针对智慧运维系统在维护机房安全时难以稳定检测机房工作人员的行为,导致出现安全隐患的问题,提出一种改进YOLOv8的行为检测算法。首先,设计一种自适应空间权重卷积模块来改进原C2f模块,提升网络对多尺度特征的获取能力;其次,提出多残差可变形卷积模块来增强算法对不规则空间特征的学习能力,并将其融入颈部网络中进一步提升对机房工作人员行为的检测精度;然后,针对当前机房图像数据集缺少的问题,从现有媒体中收集和标注相关图像,并使用迁移学习在现有训练权重基础上进一步调试优化;最后,引入Wise-IoU损失函数解决自建数据集中低质量示例对训练结果的影响。实验结果表明,改进后的算法在标准NTU RGB+D数据集的测试精度为87.84%,优于对比算法;在真实机房的测试中相较于原YOLOv8,准确度和召回率分别提高了13.24%和10.47%,参数量降低了18.07%。

关键词: 智慧运维, 机房安全, 行为检测, YOLOv8, 可变形卷积, 迁移学习

Abstract:  Aiming at the problem that the intelligent operation and maintenance system is difficult to stably detect the behavior of computer room staff when maintaining the security of the computer room, leading to potential safety hazards, an improved YOLOv8 behavior detection algorithm is proposed. Firstly, an adaptive spatial weight convolution module is designed to improve the original C2f module and enhance the network’s ability to acquire multi-scale features. Secondly, a multi-residual deformable convolution module is proposed to enhance the algorithm’s ability to learn irregular spatial features, and it is integrated into the neck network to further improve the detection accuracy of computer room staff behavior. Then, aiming at the problem of the lack of current computer room image datasets, relevant images are collected and labeled from existing media, and transfer learning is used to further debug and optimize based on existing training weights. Finally, the Wise-IoU loss function is introduced to solve the impact of low-quality examples in the self-built dataset on training results. Experiment results show that the improved algorithm achieves a test accuracy of 87.84% on the standard NTU RGB+D dataset, which is superior to the comparison algorithm; compared with the original YOLOv8 in real computer room tests, the accuracy and recall rate are improved by 13.24% and 10.47%, respectively, and the parameter quantity is reduced by 18.07%.

Key words: intelligent operation and maintenance, computer room security, behavior detection, YOLOv8, deformable convolution, transfer learning

中图分类号: