关键词: 细粒度图像分类, 注意力机制, 残差网络

Abstract: Aiming at the  characteristics of  subtle, uneven, imperceptible inter-class differences between classes and real-world data distribution in  fine-grained image classification, we proposed a fine-grained image classification model based on attention mechanism. Firstly, the preliminary feature extraction of the image was carried out  by introducing the fusion of a two-way channel attention and residual network. Secondly,  the multi-head self-attention mechanism was applied to extract fine-grained relationships between  deep feature data. Thirdly, the training of loss function measurement system was designed by combining cross entropy loss and center loss. The experimental results show that the test accuracy of the model on two standard datasets 102 Category Flower and CUB200-2011 is  94.42% and 89.43%, respectively. Compared with other mainstream classification models, the classification effect is better.

Key words: fine-grained image classification, attention mechanism, residual network