公开(公告)号：CN115170579A

公开(公告)日：2022-10-11

申请号：CN202211098621.9

申请日：2022-09-09

Applicant: 之江实验室

Inventor： 郑涛 ,  宋伟 ,  朱天成 ,  朱世强 ,  陈贤雷

IPC: G06T7/10 ,  G06N3/04 ,  G06N3/08

Abstract: 本发明属于图像识别技术领域，涉及一种金属锈蚀图像分割方法和装置，该方法包括：建立金属锈蚀图像数据集，将金属锈蚀图像数据集分为训练集和测试集；构建基于U‑net网络和混合注意力机制的深度学习分割网络模型；将训练集输入所述深度学习分割模型中进行训练，得到训练好的模型；将测试集输入所述训练好的模型中进行测试，得到金属锈蚀图像的分割结果；将得到的金属锈蚀图像的分割结果与测试集对应的标签图像中的锈蚀位置进行对比，计算评价指标，对测试结果进行评价。本发明的方法，在原始U‑net网络的基础上，添加混合注意力机制，分别从通道域和空间域两个方面对锈蚀所在区域的权重进行提升，提高了金属锈蚀图像的分割精度、准确度。

公开(公告)号：CN115170579B

公开(公告)日：2023-05-12

申请号：CN202211098621.9

申请日：2022-09-09

Applicant: 之江实验室

Inventor： 郑涛 ,  宋伟 ,  朱天成 ,  朱世强 ,  陈贤雷

IPC: G06T7/10 ,  G06N3/0464 ,  G06N3/09

Abstract: 本发明属于图像识别技术领域，涉及一种金属锈蚀图像分割方法和装置，该方法包括：建立金属锈蚀图像数据集，将金属锈蚀图像数据集分为训练集和测试集；构建基于U‑net网络和混合注意力机制的深度学习分割网络模型；将训练集输入所述深度学习分割模型中进行训练，得到训练好的模型；将测试集输入所述训练好的模型中进行测试，得到金属锈蚀图像的分割结果；将得到的金属锈蚀图像的分割结果与测试集对应的标签图像中的锈蚀位置进行对比，计算评价指标，对测试结果进行评价。本发明的方法，在原始U‑net网络的基础上，添加混合注意力机制，分别从通道域和空间域两个方面对锈蚀所在区域的权重进行提升，提高了金属锈蚀图像的分割精度、准确度。

公开(公告)号：CN116109518A

公开(公告)日：2023-05-12

申请号：CN202310323813.3

申请日：2023-03-30

Applicant: 之江实验室

Inventor： 宋伟 ,  朱天成 ,  郑涛 ,  朱世强 ,  施浩磊

IPC: G06T5/00 ,  G06T7/00 ,  G06T7/11 ,  G06T7/194 ,  G06N3/08 ,  G06N3/0464

Abstract: 本发明公开了一种金属锈蚀图像的数据增强、分割方法及装置，通过获取金属锈蚀的图像进行切分，得到多个切分后的子图，再对每一张子图进行增强操作，将切分的子图重新组合成与原始金属锈蚀图像尺寸一致的新图像，从而实现图像数据的增强，再将新图像用于金属锈蚀分割模型的训练，并通过训练好的模型对待分割的图像进行金属锈蚀分割，通过本发明对金属构建锈蚀数据集进行数据扩充，不仅增加了图像的数量，而且增加了金属锈蚀图像的多样性，有助于网络充分训练，使网络模型的泛化能力以及分割性能得到提升。