基于融合度量策略的小样本金属缺陷识别研究
DOI:
CSTR:
作者:
作者单位:

1.西南石油大学机电工程学院石油天然气装备教育部重点实验室 成都 610500; 2.四川省特种设备检验研究院 成都 610100; 3.西南石油大学石油天然气装备教育部重点实验室 成都 610500

作者简介:

通讯作者:

中图分类号:

TP391

基金项目:

四川省市场监督管理局科技计划项目 (SCSJZ2022007)资助四川省市场监督管理局科技计划项目 (SCSJZ2022007)、南充市-西南石油大学市校科技战略合作项目(SXQHJH027)


Research on few-shot learning of metal defect recognition based on fusion distribution metric strategy
Author:
Affiliation:

1.School of Mechanical and Electrical Engineering, Southwest Petroleum University,Chengdu 610500, China; 2.SIchuan Special Equipment Inspection Institute,Chengdu 610100, China; 3.Key Laboratory of Petroleum and Natural Gas Equipment of Ministry of Education,Southwest Petroleum University,Chengdu 610500,China

Fund Project:

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

    针对目前金属表面缺陷分类,数据稀缺且标注步骤繁琐昂贵的问题,将小样本度量学习引入金属表面缺陷分类中,提出了一种小样本分布度量网络模型FDM-FSL:用信息更加丰富的细节描述子来表征图像特征,并通过空间注意力机制筛选获得更具判别力的描述子信息,最后引入融合KL散度和EMD距离的图像到类的度量方式以考虑查询集和支持集类别的分布一致性。实验结果表明,提出的网络模型在小样本数据集MiniImageNet上拥有更加优良的度量能力,5类5样本下平均识别精度相较经典的RelationNet、CovaMNet、DN4算法识别准确率提高了6.34%、5.78%、1.25%。在金属缺陷数据集NEU-DET上5类5样本平均识别准确率分别提高了2.87%、3.34%、2.5%。

    Abstract:

    In view of the current metal surface defects classification, data is scarce and tagging process is cumbersome and expensive. This paper introduces the few-shot learning into the metal surface defect classification, proposes a few-shot learning network model with more informative detail descriptor to represent image features: Through adding spartial attention mechanism to screen local information and introducing the fusion measurement to class metric. Experiment results show that our model has better metric effect on MiniImageNet.We gains 6.34%, 5.78% and 1.25% improvements over RelationNet, CovaMNet and DN4 algorithms on the 5-way 5-shot task.The average accuracy of 5-way 5-shot on NEU-DET was improved by 2.87%, 3.34%, and 2.5% respectively.

    参考文献
    相似文献
    引证文献
引用本文

唐东林,刘铭璇,何媛媛,杨洲,程衡,丁超.基于融合度量策略的小样本金属缺陷识别研究[J].电子测量技术,2023,46(4):107-113

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-02-22
  • 出版日期:
文章二维码