• Laser & Optoelectronics Progress
  • Vol. 59, Issue 22, 2210002 (2022)
Cong Wu, Zhiqiang Guo, and Jie Yang*
Author Affiliations
  • College of Information Engineering, Wuhan University of Technology, Wuhan 438300, Hubei , China
  • show less
    DOI: 10.3788/LOP202259.2210002 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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