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

Abstract

The purpose of material recognition is to identify the main objects and their material categories in natural material images.Aiming at the problem of low recognition accuracy caused by the lack of data in material image data sets and the difficulty of manually labeling local texture regions,a material recognition method based on attention mechanism and deep convolutional neural network is proposed.The core of the method is material recognition deep convolutional neural network (MaterialNet).MaterialNet uses the deep residual network to extract the features of the image,and introduces the attention mechanism by the proposed cascaded atrous spatial pyramid pooling method,so that the network can adaptively focus on the key areas containing texture features through end-to-end training,so as to effectively identify the local texture features of materials.Based on the FMD material datasets,the experimental results show that the overall identification accuracy of MaterialNet is 82.3%,which is 7.2% and 4.5% higher than the current mainstream B-CNN and CNN+FV material identification methods,respectively.The recognition accuracy of MaterialNet is high for a variety of materials,and it has the advantages of less parameters and less calculation.

Keywords

• attention mechanism|deep convolutional neural network|atrous convolution|spatial pyramid pooling