├── .gitignore ├── .idea ├── .gitignore ├── Fault diagnosis Demo.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE ├── README.ch.md ├── README.md ├── datasets ├── __init__.py ├── datasets_loader.py ├── dds_datasets.py ├── hnu_datasets.py ├── preprocess.py └── xjtu_datasets.py ├── helper ├── create.py ├── loops.py ├── util.py └── util_dynamic.py ├── models ├── backbone │ ├── Image │ │ └── 2D.py │ └── Signal │ │ ├── MCSwin.py │ │ ├── __init__.py │ │ ├── convformerv1.py │ │ ├── convformerv2.py │ │ ├── crossvit.py │ │ ├── ehcnn.py │ │ ├── localvit.py │ │ ├── maxvit.py │ │ ├── mobilev2.py │ │ ├── nat.py │ │ ├── resnet.py │ │ ├── uniformer.py │ │ ├── vgg.py │ │ ├── vit.py │ │ └── wrn.py └── loss │ ├── Jsd_cross_entropy.py │ ├── enhanced_cross_entropy.py │ └── soft_cross_entropy.py ├── png ├── DDS.jpg ├── Figure_XJTUGearbox.jpg └── 湖大试验台.jpg ├── requirement.txt ├── train.py └── train_dynamic.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/Fault diagnosis Demo.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/Fault diagnosis Demo.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/LICENSE -------------------------------------------------------------------------------- /README.ch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/README.ch.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/datasets_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/datasets_loader.py -------------------------------------------------------------------------------- /datasets/dds_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/dds_datasets.py -------------------------------------------------------------------------------- /datasets/hnu_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/hnu_datasets.py -------------------------------------------------------------------------------- /datasets/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/preprocess.py -------------------------------------------------------------------------------- /datasets/xjtu_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/datasets/xjtu_datasets.py -------------------------------------------------------------------------------- /helper/create.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/helper/create.py -------------------------------------------------------------------------------- /helper/loops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/helper/loops.py -------------------------------------------------------------------------------- /helper/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/helper/util.py -------------------------------------------------------------------------------- /helper/util_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/helper/util_dynamic.py -------------------------------------------------------------------------------- /models/backbone/Image/2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Image/2D.py -------------------------------------------------------------------------------- /models/backbone/Signal/MCSwin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/MCSwin.py -------------------------------------------------------------------------------- /models/backbone/Signal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/__init__.py -------------------------------------------------------------------------------- /models/backbone/Signal/convformerv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/convformerv1.py -------------------------------------------------------------------------------- /models/backbone/Signal/convformerv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/convformerv2.py -------------------------------------------------------------------------------- /models/backbone/Signal/crossvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/crossvit.py -------------------------------------------------------------------------------- /models/backbone/Signal/ehcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/ehcnn.py -------------------------------------------------------------------------------- /models/backbone/Signal/localvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/localvit.py -------------------------------------------------------------------------------- /models/backbone/Signal/maxvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/maxvit.py -------------------------------------------------------------------------------- /models/backbone/Signal/mobilev2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/mobilev2.py -------------------------------------------------------------------------------- /models/backbone/Signal/nat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/nat.py -------------------------------------------------------------------------------- /models/backbone/Signal/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/resnet.py -------------------------------------------------------------------------------- /models/backbone/Signal/uniformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/uniformer.py -------------------------------------------------------------------------------- /models/backbone/Signal/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/vgg.py -------------------------------------------------------------------------------- /models/backbone/Signal/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/vit.py -------------------------------------------------------------------------------- /models/backbone/Signal/wrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/backbone/Signal/wrn.py -------------------------------------------------------------------------------- /models/loss/Jsd_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/loss/Jsd_cross_entropy.py -------------------------------------------------------------------------------- /models/loss/enhanced_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/loss/enhanced_cross_entropy.py -------------------------------------------------------------------------------- /models/loss/soft_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/models/loss/soft_cross_entropy.py -------------------------------------------------------------------------------- /png/DDS.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/png/DDS.jpg -------------------------------------------------------------------------------- /png/Figure_XJTUGearbox.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/png/Figure_XJTUGearbox.jpg -------------------------------------------------------------------------------- /png/湖大试验台.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/png/湖大试验台.jpg -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/requirement.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/train.py -------------------------------------------------------------------------------- /train_dynamic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bighan123/HNUIDG-Fault-Diagnosis-/HEAD/train_dynamic.py --------------------------------------------------------------------------------