一种基于WOA-VMD-YOLOv8的滚动轴承故障诊断方法

本发明公开一种基于WOA‑VMD‑YOLOv8的滚动轴承故障诊断方法，包括数据采集、数据预处理、将故障样本数据集按比例划分为训练集、测试集与验证集；搭建WOA‑VMD‑YOLOv8模型；该模型包括鲸鱼优化算法WOA、VMD分解以及YOLOv8模型；训练WOA‑VMD‑YOLOv8模型，得到基于WOA‑VMD‑YOLOv8的滚动轴承故障诊断模型；将待诊断的滚动轴承的声信号输入基于WOA‑VMD‑YOLOv8的滚动轴承故障诊断模型即可输出故障类型。本发明利用WOA优化VMD分解，并通过引入相关峭度值、加权能量熵对IMF分量进行选择，结合YOLOv8模型，提高滚动轴承故障诊断的效率与准确率。