学习目标:
四月完成第一篇基于深度学习的故障诊断小论文
目前方向,ResNeXt,transformer,ConvNeXt
学习内容:
数据集
西储大学数据集,学校自有DDS数据集
数据处理
数据的一维处理
师兄给的西储大学10分类csv格式数据集,也可以自己制作
数据的二维处理
目前可将一维mat转为二维的灰度图
神经网络
ResNext采坑过程
跑西储大学1d数据集10分类已跑通,但是整个过程数据量太大,后期loss值不稳定(已解决)
原本使用全局平均池化层来抽平数据,数据总是在最后跑的过程中loss值会突然上升并准确度下降,结果不收敛,可能是过拟合了,所以放弃全局平均池化并重新使用dense+alphadropout来进行最后的分类,数据成功收敛,准确度和loss值也稳定了,但目前还有一点瑕疵是是数据量略大
ConvNext采坑过程
tf2.5有dw2d卷积,但是没有1d的,后查询官网tf2.9才有dw1d卷积,纠结是自己把原来的分组卷积改成dw卷积拉到还是再装个一个2.9,装环境真滴要命
还好2.9与2.5使用相同的cuda环境,成功安装2.9,后也用西储数据集跑通了,但是目前没有什么