高级检索+

基于GSBS-YOLOv8的棉花病虫害轻量化检测方法

Detecting cotton pests and diseases using lightweight GSBS-YOLOv8

  • 摘要: 针对棉田自然场景下病虫害目标尺度差异大、小目标特征弱及边缘端部署受限等问题,该研究提出一种轻量化棉花病虫害检测模型GSBS-YOLOv8。模型在骨干网络和特征融合网络中引入GSConv与Ghost C2f模块,以降低模型参数量和计算复杂度;在颈部网络中引入双向特征金字塔网络(Bidirectional feature pyramid network, BiFPN),并采用Shape-IoU损失函数优化边界框回归,以增强多尺度特征表达和目标定位能力。试验结果表明,GSBS-YOLOv8的平均精度均值(mean average precision, mAP)提高1.0个百分点,参数量由3.01×106降至1.19×106,浮点运算量(floating point operations, FLOPs)由8.2 G降至4.7 G,模型体积由5.8 MB降至2.7 MB;图形处理器(graphics processing unit, GPU)环境下推理速度为101.3 帧/s,在Jetson Xavier NX平台经TensorRT加速后达到43.7 帧/s。结果表明该模型在保证检测精度的同时显著压缩了模型规模,可为棉田病虫害边缘端智能监测提供方法支持。

     

    Abstract: Scale variation of disease and pest targets has caused the weak feature representation of small objects, thus limiting edge deployment in natural cotton fields. In this study, a lightweight cotton disease and pest detection model, GSBS-YOLOv8, was proposed using you only look once version 8 (YOLOv8). Detection accuracy was improved under field conditions, while substantially reducing model complexity, computational burden, and deployment cost. In the model design, GSConv and Ghost C2f modules were introduced into the backbone network and feature fusion to replace part of the conventional convolutional structure. Thereby, parameter redundancy was reduced for feature reuse, thus lowering computational overhead without weakening semantic extraction. A bidirectional feature pyramid network (BiFPN) was embedded into the neck to strengthen the bidirectional propagation and weighted fusion of shallow spatial information and deep semantic information. The fine-grained cues of small disease spots and pests were more effectively retained at different imaging distances. In addition, the Shape-IoU loss was adopted for bounding-box regression to enhance geometric matching between predicted boxes and target contours. Localization robustness was improved for small, slender, and irregular targets under complex canopy backgrounds. Experimental results showed that the improved model achieved a 1.0 percentage point increase in mean Average Precision (mAP) over the baseline YOLOv8 model, indicating that the lightweight reconstruction preserved the high recognition of multi-scale disease and pest targets. Meanwhile, the number of model parameters decreased from 3.01 to 1.19 million, with a reduction of 60.5%. While floating point operations (FLOPs) decreased from 8.2 to 4.7 G, with a reduction of 42.7%. The model size was compressed from 5.8 to 2.7 MB, thus representing a reduction of 53.4%. Storage and transmission were markedly lowered for the feasible deployment on resource-constrained hardware platforms. The accuracy and compression efficiency were improved simultaneously. The optimal architecture was enhanced to achieve feature representation efficiency to alleviate the common trade-off between lightweight design and detection performance. In the inference evaluation, the model reached 101.3 frames per second on a graphics processing unit (GPU), indicating strong real-time performance in a high-throughput computing environment. The inference speed still reached 43.7 frames per second after TensorRT acceleration on the Jetson Xavier NX platform. Efficient execution maintained under embedded conditions, fully meeting the practical requirement of real-time field monitoring. Lightweight convolution, ghost feature, bidirectional multi-scale fusion, and regression loss can effectively balance accuracy and efficiency for the small-object perception and localization performance. The substantial decline in detection was avoided with accompanying model compression. Therefore, GSBS-YOLOv8 can provide an effective, lightweight solution for intelligent edge-side monitoring of cotton diseases and pests in natural field environments. This finding can offer technical support for subsequent applications in rapid field scouting, precision prevention and control, and low-power agricultural vision.

     

/

返回文章
返回