IEEE Access (Jan 2024)

Fine-Grained Butterfly Classification Based on Multi-Feature Enhancement

  • Hong Jin,
  • Ke Sha,
  • Xiaolan Xie

DOI
https://doi.org/10.1109/ACCESS.2024.3362073
Journal volume & issue
Vol. 12
pp. 29194 – 29203

Abstract

Read online

Due to the challenges caused by the complexity and repetitiveness of discriminative features in fine-grained butterfly images, this paper proposes a fine-grained butterfly classification model based on multi-feature enhancement and multi-scale fusion to improve the accuracy of fine-grained butterfly classification. We use ResNet50 as the base network to extract features at different scales of the image. Next, the discriminative features are enhanced using spatial attention. The enhanced discriminative features are fed into the next stage of the base network, and the above operation is repeated to obtain the enhanced discriminative features at different scales. We then use the channel covariance attention module to eliminate the extra noise introduced during multi-feature enhancement. Finally, the final feature map is obtained by fusing the complementary information of different scale feature maps using the multi-scale fusion module. Ultimately, the method used in this paper achieved 96.896% accuracy on a dataset of 10 classes of fine-grained butterfly classification, and the comparison experiments proved that the method used in this paper outperforms the mainstream fine-grained visual classification methods.

Keywords