基于 VGG16 迁移学习的轴承故障诊断方法代码(python代码,带有数据集,可以直接运行)

2023-11-09

​运行效果:基于 VGG16 迁移学习的轴承故障诊断方法代码(python代码,带有数据集,可以直接运行)_哔哩哔哩_bilibili

这个项目是复现的这篇文章:基于VGG16迁移学习的轴承故障诊断方法 - 中国知网

可以结合这篇文章理解本次项目 

代码运行库要求:TensorFlow>=2.4.0即可(可以远程协助运行起来代码)

代码文件夹解释:

第一级目录:

以0HP文件夹为例,打开后的子文件序列,1HP/2HP/3HP子文件类似,creat_picture.py是生成图片程序

0HPimages文件夹装的生成图片

VGG16文件夹装的VGG16模型对各种负载下的实验程序

CNN文件夹装的对比模型普通CNN对各种负载下的实验程序

  1. 效果图(原文采用十次迭代,这里也用十次迭代,因为采用了VGG16的预训练参数,所以测试集效果非常好):

训练集和测试集来自同一负载:

训练集和测试集来自不同负载:

对比模型CNN效果较差 ,只能将迭代次数改为100,方能稳定些,但是效果还是没有VGG16模型好

训练集和测试集来自不同负载

​对代码感兴趣的可以关注最后一行


from PIL import Image
import numpy as np
import tensorflow as tf
from keras.utils import np_utils
#代码和数据的压缩包https://mbd.pub/o/bread/ZJeXk55w

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

基于 VGG16 迁移学习的轴承故障诊断方法代码(python代码,带有数据集,可以直接运行) 的相关文章

随机推荐