电子科技大学学报2024，Vol.53Issue(1)：76-83,8.DOI:10.12178/1001-0548.2022373
先验知识辅助的金属涂层损伤分割方法
Knowledge-Driven Metal Coating Defect Segmentation
摘要
Abstract
Automatic recognition of metal coating defects has significant value in realistic applications.As deep learning makes breakthrough in surface defect segmentation for a variety of materials,most of deep convolutional neural network segmentation models are trained in an end-to-end manner.However,it is difficult to exploit prior knowledge about metal coating defects in end-to-end deep learning and adapt to the variable scale of the defects and the limited training data.This paper proposes a defect segmentation algorithm based on prior knowledge about metal coating defects to unify U-Net,a deep learning segmentation model for automatic metal coating defect recognition.This anomaly segmentation is based on Hue channel distribution and edge response.Being trained in a knowledge driven manner,the model can exclude outliers from training data and effectively avoid over-fitting.On a metal coating defect image dataset with four defect types,including crack,blister,rusting and flaking,the proposed method achieves 81.24% mIoU,which is advantageous over end-to-end deep learning.The experiment shows that knowledge-driven model can boost the performance of deep learning models in metal coating defect segmentation.关键词
深度学习/损伤识别/图像分割/先验知识Key words
deep learning/defect recognition/image segmentation/prior knowledge分类
计算机与自动化引用本文复制引用
谢洲洋,舒畅,傅彦,周俊临,蒋家玮,陈端兵..先验知识辅助的金属涂层损伤分割方法[J].电子科技大学学报,2024,53(1):76-83,8.基金项目
国家自然科学基金(61673085) （61673085）
西南技术工程研究所合作基金(HDHDW5902010301) （HDHDW5902010301）
