Laser & Optoelectronics Progress, Volume. 59, Issue 22, 2210002(2022)

Algorithm for Plug Seedling Classification Based on Improved Attention Mechanism Residual Network

Cong Wu, Zhiqiang Guo, and Jie Yang*
Author Affiliations
  • College of Information Engineering, Wuhan University of Technology, Wuhan 438300, Hubei , China
  • show less

    A residual network based on dual-channel attention mechanism is designed to address the difficulty in extracting single and segmented seedling dataset images from the channel attention mechanism feature of the SENet network, which integrates the channel attention mechanism and spatial attention. The mechanism module can obtain the channel and spatial dimension feature weights simultaneously to enhance the feature learning ability of the network. To address the problem of missing the target in the segmented sample data, a random erasure method is proposed. Experiments on the self-made plug seedling Plant_seed dataset demonstrate that the improved network ResNet34+CBAM_basic_conv, which introduces the attention mechanism module between the ResNet34 network residual module and the conv*_x module, reaches the optimal accuracy of 93.8%. The error rate of the model classification drops after some images in the dataset are randomly erased, demonstrating the excellent performance of the proposed method.

    Tools

    Get Citation

    Copy Citation Text

    Cong Wu, Zhiqiang Guo, Jie Yang. Algorithm for Plug Seedling Classification Based on Improved Attention Mechanism Residual Network[J]. Laser & Optoelectronics Progress, 2022, 59(22): 2210002

    Download Citation

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

    Category: Image Processing

    Received: Aug. 23, 2021

    Accepted: Sep. 24, 2021

    Published Online: Sep. 23, 2022

    The Author Email: Yang Jie (jieyang@whut.edu.cn)

    DOI:10.3788/LOP202259.2210002

    Topics