基于 DL 的智能诊断基准
深度学习 旋转机械智能诊断算法: 开源 基准研究 ISA Transactions arXiv 由 Zhibin Zhao、Tianfu Li 和 Jingyao Wu 编写。
更正
- 2020.05.14, 我们修改了与 XJTU 数据集相关的数据加载器中的错误。
指南 该项目仅提供基准（下限）精度和统一的智能故障诊断库，后者保留了扩展的接口，供每个人自行加载自己的数据集和模型，以进行新的研究。同时，所有实验均在 Windows 10 和 Pytorch 1.1 环境下运行，使用配备 Intel Core i7-9700K、GeForce RTX 2080Ti 和 16G RAM 的计算机。
要求
- Python 3.7
- Numpy 1.16.2
- Pandas 0.24.2
- 酱
- tqdm 4.31.1
- sklearn 0.21.3
- Scipy 1.2.1
- opencv-Python 4.1.0.25
- PyWavelets 1.0.2
- pytorch >= 1.1
- torchvision >= 0.40
数据集
包
此仓库的组织结构如下: - AE_Datasets 包含用于 AE 模型的不同数据集的加载器。 - CNN_Datasets 包含用于 MLP、CNN 和