高级检索+

基于改进YOLO11-seg的轻量化病虫害分割模型

Lightweight pest segmentation model based on improved YOLO11-seg

  • 摘要: 针对病虫害分割中存在的背景复杂、边界模糊、现有算法计算量较大等问题,该研究提出一种改进YOLO11-seg的轻量化病虫害实例分割网络。首先,在特征提取阶段,使用C3k2_FAC替换原始的C3k2模块,降低计算成本并强化通道特征响应;其次,设计Focaler-Shape IoU作为新的损失函数,提高对复杂形状和边缘区域的分割精度;再次,采用LAMP通道剪枝方法进一步压缩模型大小和计算量;最后,利用通道级知识蒸馏进行精度恢复,提高模型的分割性能。试验结果表明,在自建的病虫害数据集上,该改进模型在mask水平的mAP50和mAP50-95分别为82.7%和55.7%,模型大小仅有6.3MB,参数量和计算量较原始模型分别降低了70.3和35.9个百分点,在分割精度和轻量化方面相较于其他先进算法均有明显的优势。该研究可为农业病虫害智能化监测提供技术支持。

     

    Abstract: Accurate segmentation is often required for the early diagnosis and precise prevention of the crop diseases and pests. However, the existing segmentation of the pests and diseases has been limited to the complex background, fuzzy boundary, and large computation. In this study, a lightweight network with YOLO11-seg was proposed for the pest and disease segmentation. Firstly, the original C3k2 module was replaced with C3k2_FAC in the feature extraction stage, in order to reduce the computational cost for the channel feature response; Secondly, the Focaler-Shape IoU loss function was designed to improve the segmentation accuracy for the complex shapes and edge regions; Thirdly, the LAMP channel pruning was used to further compress the model size and computation; Finally, the channel-level knowledge distillation was utilized for the accuracy recovery, in order to improve the segmentation performance of the model. The dataset consisted of two samples, namely diseases and pests. Among them, the disease samples were collected from the PlantSeg dataset, from which 10 types of disease examples of 5 cash crops, such as corn, wheat, and soybean, were selected as research objects. The pest samples were mainly collected from the network and on-site collection, and then 1 600 pest images were captured. The original dataset of the pests and diseases was expanded to 5 863 using multiple image enhancement techniques, including affine changes, in order to avoid the uneven distribution of data samples and too few samples. After that, the dataset was randomly divided into 4690 training sets, 586 verification sets, and 587 test sets, according to the ratio of 8:1:1. A series of experiments were was conducted using the RTX3090 computing platform with training batches of 32 and 150 iterations on the self-constructed pest and disease dataset. The experimental results show that the optimal performance was achieved after replacing C3k2 in the backbone network with C3k2_FAC. The model parameters and computational load were reduced by 0.7M and 1.3G, respectively, while the mAP increased by 0.4 percentage points. Therefore, the C3k2_FAC module was improved the feature extraction for the high computational efficiency. The Focaler-Shape IoU loss function was introduced, with the mAP50 and mAP50-95 of 82.7% and 55.0%, respectively. Furthermore, the model size was reduced to 6.0MB, after the LAMP channel pruning was applied with two improvements, significantly decreasing the number of parameters and computational load, although the mAP dropped by 0.5 percentage points. The mAP50 and mAP50-95 increased by 0.6 and 0.5 percentage points, respectively, after knowledge distillation, in order to balance the model parameters and computational load. The mAP50 and mAP50-95 of the improved model were 82.7% and 55.7%, respectively, at the mask level, compared with the original model. The size of the model was only 6.3MB, while the number of parameters and the computational volume were reduced by 70.3 and 35.9 percentage points, respectively. There were the outstanding advantages in the segmentation accuracy and lightweight performance, compared with the rest advanced algorithms, such as the Mask R-CNN, SOLOv2, YOLACT, YOLOv8s-seg, and YOLOv9-C. Nine images of the crop disease and pest were selected from the test set. For instance, the segmentation and visual display demonstrated the better recognition on of the pest and disease targets. Leaf diseases were identified to successfully segment on the contours of the lesions, whether they were the single large lesions or multiple small ones. Additionally, the specific shapes and locations of small pests can be expected to accurately distinguish for the high accuracy even amidst relatively cluttered backgrounds. The high accuracy with the lightweight was achieved in the segmentation and identification of the pests. This finding can also provide the technical support to intelligently monitor the agricultural pests and diseases.

     

/

返回文章
返回