This project contains various scripts and models for training and testing on a specific dataset. Below is a brief description of the directory structure and the purpose of each script.
dataset/
: The folder where the dataset is stored. To obtain the dataset, please contact the author via email at [email protected].main/
: Contains the main training and testing scripts for different models and experiments.models/
: Contains the model architectures used in the experiments.utils/
: Contains utility scripts for data processing and other helper functions.
run_predic.sh
: Training script for GAN models.run_predic_nogan.sh
: Training script for MODAL-DRN models.run_predic_nogan_bls.sh
: Training script for MODAL-DRN-BL models.run_predic_stressnet.sh
: Training script for StressNet, SCSNet, and Inbetween models.run_predic_nogan_resnet.sh
: Training script for ablation experiments.
test_predic.sh
: Testing script for GAN models.test_predic_bls.sh
: Testing script for MODAL-DRN-BL models.test_predic_nogan.sh
: Testing script for MODAL-DRN models.test_predic_nogan_bls.sh
: Testing script for MODAL-DRN-BL models.test_predic_nogan_resnet.sh
: Testing script for ablation experiments.
If you have any questions about the code files, please contact the author at [email protected].
lf you find this repo helpful, please cite the following paper:
@inproceedings{zeng-etal-2024,
title = "MODAL-DRN-BL: A framework for modal analysis based on dilated residual broad learning networks",
author = "Zeng, Peijian and Lin Nankai and Shun Li and Lin Jianghao and Yang Aimin and Hou Liheng",
booktitle = "Journal of Computing and Information Science in Engineering",
year = "2024",
publisher = "American Society of Mechanical Engineers" }