运行效果:基于 VGG16 迁移学习的轴承故障诊断方法代码(python代码,带有数据集,可以直接运行)_哔哩哔哩_bilibili
这个项目是复现的这篇文章:基于VGG16迁移学习的轴承故障诊断方法 - 中国知网
可以结合这篇文章理解本次项目
代码运行库要求:TensorFlow>=2.4.0即可(可以远程协助运行起来代码)
代码文件夹解释:
第一级目录:
以0HP文件夹为例,打开后的子文件序列,1HP/2HP/3HP子文件类似,creat_picture.py是生成图片程序
0HPimages文件夹装的生成图片
VGG16文件夹装的VGG16模型对各种负载下的实验程序
CNN文件夹装的对比模型普通CNN对各种负载下的实验程序
-
效果图(原文采用十次迭代,这里也用十次迭代,因为采用了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