• Laser Journal
  • Vol. 45, Issue 4, 165 (2024)
HOU Xiangning1,2, ZHAO Jinwei3, HUANG Xiaobin1,2, and JIANG Weicheng1,2
Author Affiliations
  • 1College of Engineering and Technical, Cheng du University of Technology, Leshan 614000, China
  • 2Southwestern Institute of Physics, Chengdu 610200, China
  • 3School of Computer Science and Engineering, Xi’an University of Technology, Xi’an 710048, China
  • show less
    DOI: 10.14016/j.cnki.jgzz.2024.04.165 Cite this Article
    HOU Xiangning, ZHAO Jinwei, HUANG Xiaobin, JIANG Weicheng. Flower classification based on bilinear RepVGG attention network[J]. Laser Journal, 2024, 45(4): 165 Copy Citation Text show less

    Abstract

    To further improve the accuracy of flower classification, a new network model was proposed based on bilinear convolutional neural network, RepVGG and attention mechanism. Firstly, RepVGG network was used to replace the original feature extraction network VGG to improve the ability to extract the main features of flowers. Then, channel attention and spatial attention mechanisms were introduced into the two RepVGG networks respectively, and the high-dimensional bilinear features generated by the cross-product of the two RepVGG networks were used to extract the fine-grained features of flowers. Finally, the RepVGG layers are transformed into single-way structures by structure reparameterization to improve the speed of model reasoning. Experimental results show that on the enhanced Oxford-102 data set, the inference speed and classification accuracy of the new model are greatly improved compared with the original model and the common model, and the classification accuracy is also improved compared with that before the introduction of attention.