Laser Journal, Volume. 45, Issue 4, 165(2024)

Flower classification based on bilinear RepVGG attention network

HOU Xiangning1...2, ZHAO Jinwei3, HUANG Xiaobin1,2, and JIANG Weicheng12 |Show fewer author(s)
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

    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.

    Tools

    Get Citation

    Copy Citation Text

    HOU Xiangning, ZHAO Jinwei, HUANG Xiaobin, JIANG Weicheng. Flower classification based on bilinear RepVGG attention network[J]. Laser Journal, 2024, 45(4): 165

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Sep. 14, 2023

    Accepted: Nov. 26, 2024

    Published Online: Nov. 26, 2024

    The Author Email:

    DOI:10.14016/j.cnki.jgzz.2024.04.165

    Topics