一种基于改进后EfficientNetV2模型的桥梁混凝土缺陷多标签分类方法

    公开(公告)号:CN119006928A

    公开(公告)日:2024-11-22

    申请号:CN202411234860.1

    申请日:2024-09-04

    Abstract: 本发明公开了一种基于改进后EfficientNetV2模型的桥梁混凝土缺陷多标签分类方法,属于混凝土缺陷分类技术领域。本发明通过在原始的EfficientNetV2网络中加入SCARM模块,实现对网络提取的特征图进行空间和通道上的冗余信息的去除,以减少模型训练过程中的冗余信息,并通过注意力机制促进代表性特征的学习;鉴于数据集中的图像具有不同的尺寸和高宽比,在分类层之前采用了SPPF模块进行多尺度特征融合,可以帮助网络提取各种尺寸的空间特征信息,从而提高模型对物体变形的鲁棒性;为了解决数据集中正负样本不平衡的问题,采用了不对称损失函数进行训练,能够动态调整权重和硬阈值,以减少简单负样本的影响,并且可以过滤可能被错误标记的样本。

Patent Agency Ranking