基于改进Deeplabv3+的磁瓦表面缺陷分割
DOI:
CSTR:
作者:
作者单位:

安徽工程大学

作者简介:

通讯作者:

中图分类号:

基金项目:

安徽省教育厅重大项目(KJ2020ZD39);安徽省高等学校省级质量工程项目(2023cxtd057)


Magnetic tile surface defect segmentation based on improved Deeplabv3+
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对磁瓦图像中不同类型的缺陷大小不一影响分割准确率、检测模型参数量大在实际应用中难以部署、图像像素分布不均匀的问题,提出了一种基于改进Deeplabv3+的磁瓦表面缺陷分割算法。首先,在结构上,主干网络采用轻量化的MobilNetv3网络代替原模型的Xception网络,使得模型的参数和计算量保持较小以提升检测速度;其次,引入ECANet注意力机制,提升模型的特征表达能力和泛化能力;最后,采用Dice Loss和Focal Loss相结合的损失函数,有效缓解样本像素点分布不平衡对模型训练的影响。将各个改进点进行消融实验,再将改进后的Deeplabv3+与其他模型进行对比,实验结果表明,本文算法在magnetic tile surface defects数据集上平均交并比为68.25%,平均像素准确率为82.80%,准确率为79.80%,相较于原Deeplabv3+算法,平均交并比提升了8.62%,平均像素准确率提升了9.96%,准确率提升了11.52%,验证了本文算法的有效性和可行性,在工业应用中具备一定的实际应用价值。

    Abstract:

    In order to solve the problems that the size of different types of defects in magnetic tile images affects the segmentation accuracy, the detection model parameters are large and difficult to deploy in practical applications, and the image pixel distribution is uneven, a magnetic tile surface defect segmentation algorithm based on improved Deeplabv3+ was proposed. Firstly, in terms of structure, the lightweight MobilNetv3 network was used as the backbone network to replace the Xception network of the original model, so that the parameters and computational cost of the model were kept small to improve the detection speed. Secondly, the ECANet attention mechanism was introduced to improve the feature expression ability and generalization ability of the model. Finally, the loss function combining Dice Loss and Focal Loss were used to effectively alleviate the influence of sample pixel distribution imbalance on model training. The experimental results show that the average intersection union ratio of the proposed algorithm on the magnetic tile surface defects dataset was 68.25%, the average pixel accuracy was 82.80%, and the accuracy was 79.80%, compared with the original Deeplabv3+ algorithm, the average intersection union ratio was increased by 8.62%, and the average pixel accuracy was increased by 9.96%, the accuracy of the algorithm was increased by 11.52%, which verifies the effectiveness and feasibility of the proposed algorithm, and has certain application value in industrial applications.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-08-25
  • 最后修改日期:2024-12-13
  • 录用日期:2024-12-17
  • 在线发布日期:
  • 出版日期:
文章二维码
×
《电子测量与仪器学报》
财务封账不开票通知