#数据集的文件夹路径 train_image_dir = r'E:\dataset\CRACK500\train\image' train_mask_dir = r'E:\dataset\CRACK500\train\mask' valid_image_dir = r'E:\dataset\CRACK500\validation\image' valid_mask_dir =r'E:\dataset\CRACK500\validation\mask' test_image_dir = r'E:\dataset\CRACK500\te...
6.步骤s2、分别对步骤s1中路面裂缝训练集进行数据扩充,路面裂缝验证集和路面裂缝测试集不进行数据扩充,得到路面裂缝样本集,再利用步骤s1得到的路面裂缝验证集选择训练裂缝检测网络,得到训练裂缝检测网络内最优的模型,路面裂缝测试集用于测试网络模型性能; 7.步骤s3、构建引入cbam注意力机制改进型u-net卷积神经网络分割模...
首先建立永磁同步电机(PMSM)和非晶合金变压器(AMT)有限元模型,得到深度学习训练所需的数据集;然后将Transformer模块与U-Net模型结合,并引入短残差机制建立ResUnet-Transformer模型,通过预测图像的像素实现磁场预测;最后通过Targeted Dropout算法和动态学习率调整策略对模型进行优化,解决拟合问题并提高预测精度。计算实例证明,Re...