Page 61 - 无损检测2024年第六期
P. 61

任建浩,等:

              基于 1D-CNN-SVM 的钢丝绳损伤识别方法































                                          图 5  3 种类型损伤及无损伤钢丝绳的检测信号波形
                 表1 3种类型损伤及无损伤钢丝绳样本数据                           0. 000 5,momentum动量参数为0. 9,epoch为50。

                 损伤类型       样本总量       测试集数量         标签         2.3.2  试验结果分析
                0.5根断丝        300         60          0              为验证所提模型的有效性,将数据集中数据输入
                 1根断丝         300         60          1
                                                                到所提模型中的1D-CNN模型进行训练,训练过程中
                 2根断丝         300         60          2
                                                                的训练集和测试集的损失率和准确率散点图如图6所
                 无断丝          300         60          3
                                                                示,可见第30次迭代后准确率稳定在98%左右,这里
                      表2 1D-CNN-SVM模型参数设置                       的准确率关系到SVM分类的准确率,准确率越高则
                网络模型      网络结构           参数       激活    数目      后面输入到SVM分类器中的提取特征越准确。
                           Conv1      1×3,stride:1  relu  16
                            Pool1     1×2,stride:2  —   16
                           Conv2      1×3,stride:1  relu  32
                            Pool2     1×2,stride:2  —   32
                1D-CNN     Conv3      1×3,stride:1  relu  64
                            Pool3     1×2,stride:2  —   64
                            Fc1         16256      —    —
                           dropout      P=0.5     relu  —
                            Fc2          512       —    —
                 SVM     SVMClassifier  kernel='rbf '  —  —
             (卷积核数量×长度×深度)结构的卷积核,为了
              更全面深入提取到缺陷特征,使得每一层输出通道
              翻倍,每一层都会经过批量划归一层和激活层处理;
              在1D-CNN的两个全连接层中穿插一个dropout层,                        图 6  1D-CNN 模型训练过程中的准确率和损失率散点图
              使模型具有更好的鲁棒性;最后将提取到的特征与                                 为了更直观地观察模型分类效果,使用 t-分布
              对应标签输入到SVM分类器中进行缺陷识别并输                            随机近邻嵌入(t-SNE,一种用于降维和可视化高维
              出结果。模型在基于python平台的深度学习框架                          数据的技术)进行分析,将提取的高维特征映射到
              pytorch下运行,GPU(图形处理器)为RTX 3090,                   二维平面,输入层的t-SNE降维可视化结果如图7所
              操作系统为Ubantu。模型训练的超参数中,设置训                         示,可见,训练前的测试集数据分布混杂,难以对不
              练批量为16,选择SGD作为优化器,设置学习率为                          同钢丝绳损伤进行分类。
                                                                                                          27
                                                                                         2024 年 第 46 卷 第 6 期
                                                                                                  无损检测
   56   57   58   59   60   61   62   63   64   65   66