基于ResNet18和迁移学习的滚动轴承故障诊断
组合机床与自动化加工技术
页数: 5 2024-08-16
摘要: 针对基于深度学习的滚动轴承故障诊断网络层数过深模型易退化、难以收集大量故障样本的问题,提出一种基于ResNet18和迁移学习的滚动轴承故障诊断方法。首先,采用小波变换将原始振动信号转换为二维时频图像,并通过图像增强方法凸显图像包含的时频信息;其次,通过迁移学习,将在ImageNet数据集上预训练的ResNet18作为初始故障诊断模型;最后,利用轴承数据集微调网络所有参数,生成最... (共5页)