计算机科学2021,Vol.48Issue(10) :220-225.DOI:10.11896/jsjkx.200800073

基于注意力机制和深度卷积神经网络的材质识别方法

Material Recognition Method Based on Attention Mechanism and Deep Convolutional Neural Network

许华杰 杨洋 李桂兰
计算机科学2021,Vol.48Issue(10) :220-225.DOI:10.11896/jsjkx.200800073

基于注意力机制和深度卷积神经网络的材质识别方法

Material Recognition Method Based on Attention Mechanism and Deep Convolutional Neural Network

许华杰 1杨洋 2李桂兰3
扫码查看

作者信息

  • 1. 广西大学计算机与电子信息学院 南宁 530004;广西多媒体通信与网络技术重点实验室 南宁 530004
  • 2. 广西大学计算机与电子信息学院 南宁 530004
  • 3. 广西壮族自治区产品质量检验研究院 南宁530007
  • 折叠

摘要

材质识别旨在识别自然材质图像中的主要对象及其所属材料类别.针对材质图像数据集通常数据量少、人工标注局部纹理区域困难所导致的材质识别准确率低的问题,提出了一种基于注意力机制和深度卷积神经网络的材质识别方法,该方法的核心是材质识别深度卷积神经网络(MaterialNet).MaterialNet利用深度残差网络对图像进行特征提取,采用所提出的级联空洞空间金字塔池化的方式引入注意力机制,使网络可以通过端到端训练自适应地关注包含纹理特征的关键区域,从而有效识别材质的局部纹理特征.在FMD材质数据集上进行实验,结果表明,MaterialNet的总体识别准确率可达到82.3%,比当前主流的B-CNN和CNN+FV材质识别方法分别提高了7.2%和4.5%,对多种材质的识别准确率较高且具有参数量少、计算量小等优点.

关键词

注意力机制/深度卷积神经网络/空洞卷积/空间金字塔池化

引用本文复制引用

基金项目

出版年

2021
计算机科学
重庆西南信息有限公司(原科技部西南信息中心)

计算机科学

CSTPCDCSCD北大核心
影响因子:0.944
ISSN:1002-137X
参考文献量1
段落导航相关论文