DeepBDC 跑轴承故障的时频图
cub训练的模型测试
- 先是用cub数据集训练的模型,对轴承数据集测试
先看1-shot的:
![在这里插入图片描述](https://img-blog.csdnimg.cn/1053b94a24f74c64bd0088d5b0ce82e5.png
这是5-shot的:
效果感觉不错。
- 遇到的问题:
- 数据集太小了,只有10个类,之前在windows下面python画图龟速,每个类只画了30张图片。
- 用linux脚本再画一次,应该会快很多
- [ ]
miniImage训练的模型测试
- 暂时是失败的
- 报错:RuntimeError: shape ‘[100, 3, 84, 84]’ is invalid for input of size 846720
猜测原因:类别数量太少
故障诊断论文综述
结果
任务 | shots | acc |
---|---|---|
T1 | 1shots | 0.80 |
T1 | 5shots | 0.90 |
T2 | 1shots | 0.72 |
T2 | 5shots | 0.85 |