当前位置: 首页 > article >正文

轻量级网络EfficientNetB0,利用迁移学习中的微调技术进行小样本轴承故障诊断(Python代码,带有数据集,训练集集的每类只需10个样本)

1. 训练集数量(正常类别:10个,内圈故障:10个,外圈故障:10个,滚动体故障:10个)

测试集数量(正常类别:90个,内圈故障:90个,外圈故障:90个,滚动体故障:90个)

模型:EfficientNetB0(利用imagenet训练好的参数,冻结高层,只将分类器的参数利用训练集微调,利用整个模型对测试集识别)

2.代码文件夹解释: 

第一级目录(采用的是CWRU数据集,CWRU有4种负载数据集):

3. 0HP、1HP、2HP与3HP文件夹内容类似,这里以  0HP文件夹为例,可以看到0HP文件夹里有:

Inner文件夹里装的是内圈故障的.mat保存的数据

Normal文件夹里装的是正常的.mat保存的数据

Outer文件夹里装的是外圈故障的.mat保存的数据

Roller文件夹里装的是滚动体故障的.mat保存的数据

creat_picture.py文件是数据处理文件,将四种类别的一维数据变成二维图像,生成图像样本和标签,最终,每个类别下有100个图片。对应标签被保存在label.npy。

运行creat_picture.py后, 0HPimages文件夹装的生成图片

 

4.结果(运行EfficientNetB0_0HP.py进行诊断)

0HP数据集结果:

 

 

1HP数据集

2HP数据集:

 3HP数据集:

https://mbd.pub/o/bread/ZJiWlZpp​​​​​​​


http://www.kler.cn/a/15061.html

相关文章:

  • 【环境配置】ubuntu-jetson上的定时任务
  • React|bpmn.js|react-bpmn使用示例详解
  • 优化C++设计模式:用模板代替虚函数与多态机制
  • 【Android】Proxyman 抓 HTTP 数据包
  • C++ | Leetcode C++题解之第565题数组嵌套
  • uni-app快速入门(七)--组件路由跳转和API路由跳转及参数传递
  • 【接口自动化测试】selenium旗舰版Web测试理论篇
  • 听我一句劝,别去外包,干了三年,废了....
  • css中的background属性
  • BatchNormalization 介绍
  • 800字带你弄懂Http请求和响应
  • 机器人学一些知识
  • 基于趋动云的 Stable Diffusion Webui 环境搭建
  • 低调且强大——JNPF低代码平台
  • 微服务监控系统选型:Zabbix、Open-Falcon、Prometheus
  • 【LeetCode: 1143. 最长公共子序列 | 暴力递归=>记忆化搜索=>动态规划】
  • SpringBoot启动时自动创建数据库表
  • [Java List类型和Map类型想换转换案例]
  • 2023年产业基金研究报告
  • 【回眸】ChatGPT Plus(GPT4体验卡)
  • L1-6 剪切粘贴
  • 写一个加载远程js的函数,function loadJSFn(sucCb,failCb){}, 要考虑到低端浏览器的兼容性;
  • 如何基于vue实现倒计时效果
  • 字节的面试,你能扛住几道?
  • NetSuite 负库存控制功能包
  • Go分布式爬虫笔记(二十)