基于改进深度学习的风机油污识别

来源：计算机测量与控制杂志2023年第5期北京时间：

作者：李家源,曹雪虹,焦良葆,张智坚,陈 烨

单位：南京工程学院 人工智能产业技术研究院

摘要：针对风机设备油液渗漏影响风机正常运行亟需解决的对风机设备油污的识别问题，提出了一种基于改进深度学习的风机油污检测方法。基于深度学习在目标检测中的应用特点，对目标检测网络YOLOv5n（You Only Look Once v5n）进行改进，将原网络中的非极大抑制（Non Maximum Suppression，NMS）替换为Soft-NMS，降低了网络的误检率，添加CA (Coordinate Attention)注意力机制，增强了模型对目标的定位能力，改进原网络损失函数为α-IoU（Alpha- Intersection over Union）损失函数，提高了边界框检测的准确度。实验结果表明:模型平均精度提升了8.1%，查全率提高了19.1%，网络推理速度提高了28.6%。改进后的模型能准确检测风机油污，有效解决了风机实际运行中油液渗漏所带来的问题。

关键词：深度学习;风机油污;注意力机制;损失函数;非极大值抑制

基金资助：江苏省高等学校自然科学基金面上项目(21KJB120005)