Semantic segmentation network based on attention mechanism for wheat FHB

Chen Peng,Ma Zihan,Zhang Jun,Xia Yi,Wang Bing,Liang Dong
DOI: https://doi.org/10.13733/j.jcam.issn.2095-5553.2023.04.020
2023-01-01
Abstract:Fusarium Head Blight(FHB) of wheat is one of the most terrible diseases that lead to wheat yield reduction. It is of great significance to carry out automatic identification research on wheat FHB. However, the traditional methods of segmentation and recognition of wheat scab under complex field background is generally carried out through threshold value, color histogram, etc., and its segmentation and recognition accuracy is poor and its generalization ability is not satisfactory. In order to quickly and accurately segment FHB scab and then effectively confirm the severity of the disease for assisting agricultural workers to carry out subsequent researches, this paper proposes a semantic segmentation network model, UNetA, based on UNet structure and attention mechanism for wheat FHB. The wheat ear pictures are augmented and then input into the convolution layer of UNetA model for extracting feature map. The Attention mechanism consists with position attention and channel attention after convolution layer to make further extraction, and then Encoder-Decoder structure with skip-connection and BN layer makes up the rest part together. The whole network utilizes cross-entropy loss with weighted parameters to balance the gap between classes and to measure the difference between predicted label and actual label. Subsequently, UNetA model is compared with the state-of-the-art methods. The experiment results show that the proposed method performs favorably against others in terms of MIoU in the same configuration and obtains an 83.90% of MIoU. Moreover, the proposed method spends 0.588 0 s time for wheat sacb segmentation, shorter than others.
What problem does this paper attempt to address?