一种基于改进U-Net的机械设备表面损伤识别方法

本发明提供一种基于改进U‑Net的机械设备表面损伤识别方法，包括：采集具有损伤特征的机械设备表面图像进行清洗和裁剪，并划分为训练集、验证集和测试集；选取训练集进行数据增强；将增强训练集输入机械设备表面损伤识别模型进行训练；机械设备表面损伤识别模型在U‑Net模型的基础上引入颜色特征提取模块，颜色特征提取模块包括1×1卷积层、批归一化层和ReLU激活函数，编码器包括依次连接的四个第一特征提取模块和一个第二特征提取模块，解码器包括依次连接的四个特征融合模块，特征融合模块和第一特征提取模块一一对应；用验证集对改进U‑Net模型进行验证；用测试集对最终的改进U‑Net模型进行测试，并统计评价指标。