Laser & Optoelectronics Progress, Volume. 59, Issue 22, 2210002(2022)
Algorithm for Plug Seedling Classification Based on Improved Attention Mechanism Residual Network
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.
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
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)