计算机技术与发展2023,Vol.33Issue(7) :167-172,180.DOI:10.3969/j.issn.1673-629X.2023.07.025

基于鉴别注意力融合的仪表细粒度分类方法

A Fine-grained Classification Method for Industrial Meters Based on Discriminant Attention Fusion

孙荣艳 李晓明
计算机技术与发展2023,Vol.33Issue(7) :167-172,180.DOI:10.3969/j.issn.1673-629X.2023.07.025

基于鉴别注意力融合的仪表细粒度分类方法

A Fine-grained Classification Method for Industrial Meters Based on Discriminant Attention Fusion

孙荣艳 1李晓明1
扫码查看

作者信息

  • 1. 太原科技大学 计算机科学与技术学院,山西 太原 030024
  • 折叠

摘要

基于视觉的仪表自动巡检读数是一项重要的研究内容;仪表的读数通常通过检测指针的位置来确定;不同规格型号的仪表,指针指向相同位置所代表的具体读数不同,因此,预先识别仪表的详细类别是进行自动读数的重要前提.为提升仪表自动读数的便捷性和准确率,提出了一种基于鉴别注意力融合的仪表细粒度分类方法.首先利用YOLOv5 对仪表表盘进行粗提取,随后使用提出的模型对表盘进行细粒度识别;无需标注显著特征,在特征提取器上添加鉴别注意力模块,补充浅层空间和位置信息对仪表细粒度分类的引导作用;生成的鉴别粒度注意力图和骨干网络最后一层输出的特征图进行双线性融合,生成特征矩阵;引入正交损失,对生成的特征矩阵进行约束处理;构建仪表细粒度分类数据集.理论分析和实验结果表明,所提仪表细粒度分类方法提高了网络对表盘鉴别粒度区域的识别能力,有效改善了仪表细粒度分类的性能,为后续仪表智能准确的读数提供了保证.

关键词

双线性融合/正交损失/类激活热力图/YOLOv5/工业仪表

Key words

bilinear fusion/orthogonal loss/class activation map/YOLOv5/industrial meter

引用本文复制引用

基金项目

国家自然科学基金(61373099)

出版年

2023
计算机技术与发展
陕西省计算机学会

计算机技术与发展

CSTPCD
影响因子:0.621
ISSN:1673-629X
参考文献量2
段落导航相关论文