A Lightweight Winter Wheat Planting Area Extraction Model Based on Improved DeepLabv3+ and CBAM

Abstract

This paper focuses on the problems of inaccurate extraction of winter wheat edges from high-resolution images, misclassification and omission due to intraclass differences as well as the large number of network parameters and long training time of existing classical semantic segmentation models. This paper proposes a lightweight winter wheat planting area extraction model that combines the DeepLabv3+ model and a dual-attention mechanism. The model uses the lightweight network MobileNetv2 to replace the backbone network Xception of DeepLabv3+ to reduce the number of parameters and improve the training speed. It also introduces the lightweight Convolutional Block Attention Module (CBAM) dual-attention mechanism to extract winter wheat feature information more accurately and efficiently. Finally, the model is used to complete the dataset creation, model training, winter wheat plantation extraction, and accuracy evaluation. The results show that the improved lightweight DeepLabv3+ model in this paper has high reliability in the recognition extraction of winter wheat, and its recognition results of OA, mPA, and mIoU reach 95.28%, 94.40%, and 89.79%, respectively, which are 1.52%, 1.51%, and 2.99% higher than those for the original DeepLabv3+ model. Meanwhile, the model’s recognition accuracy was much higher than that of the three classical semantic segmentation models of UNet, ResUNet and PSPNet. The improved lightweight DeepLabv3+ also has far fewer model parameters and training time than the other four models. The model has been tested in other regions, and the results show that it has good generalization ability. The model in general ensures the extraction accuracy while significantly reducing the number of parameters and satisfying the timeliness, which can achieve the fast and accurate extraction of winter wheat planting sites and has good application prospects.