锈蚀检测模型构建方法及锈蚀检测方法

    公开(公告)号:CN115994889B

    公开(公告)日:2025-03-25

    申请号:CN202211333214.1

    申请日:2022-10-28

    Abstract: 本发明提供了一种锈蚀检测模型构建方法及锈蚀检测方法。其中,锈蚀检测模型构建方法,包括:获取模型训练数据,模型训练数据中包括多个金属设备图像,各金属设备图像中标注有锈蚀区;基于模型训练数据,对U2Net网络模型进行模型训练,得到锈蚀检测模型,U2Net网络模型的顶层由编码器和解码器构成U型网络结构,编码器和解码器分别由多层Swin Transformer单元构成U型网络结构。通过本发明,利用Swin Transformer单元进行局部和全局特征学习,提取各阶段的多尺度特征和聚集各阶段的多层次特征,提高锈蚀检测模型检测锈蚀区域的准确性。

Patent Agency Ranking