期刊
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE
卷 122, 期 -, 页码 -出版社
PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.engappai.2023.106011
关键词
Surface defect segmentation; Sub-regions features; Mask-aware loss; Group convolution; Global Block; Spatial channel transform
In recent years, diverse detection methods have been proposed to achieve highly accurate image segmentation. However, low contrast and background noise interference still pose challenges to defect detection. Existing deep learning-based methods for defect detection mostly learn rich features on the whole image.
In recent years, diverse detection methods have been proposed to achieve highly accurate image segmentation. However, low contrast and background noise interference still pose challenges to defect detection. Existing deep learning-based methods for defect detection mostly learn rich features on the whole image. This can lead to overfitting as the parameters get bigger, and it also loses the spatial connection. In order to solve this problem, we suggest a network based on spatial sub-region feature extraction and mask-aware loss that can detect metal surface defects. In this network framework, we obtain the contextual information of each region through two streams: one branch encodes the spatial sub-region information and reduces the effect of geometric and illumination deformation under u-shape architecture. A skip-connected structure in another stream is designed to learn global information and compensate for boundary features. Next, we propose a mask-aware loss to reduce background noise interference further. Finally, we validate the effectiveness of our method on two challenging datasets, reaching 88.45% and 79.16% on the NEU-Seg and USB-Seg datasets, respectively. Also, our model is only 5.48 MB, and can achieve 172 fps for processing images with 200 x 200 size defects.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据