高级检索+

基于红外热成像和改进YOLO v5的作物病害早期识别

Early Identification of Crop Diseases Based on Infrared Thermography and Improved YOLO v5

  • 摘要: 为实现作物病害早期识别,本文提出一种基于红外热成像和改进YOLO v5的作物病害早期检测模型,以CSPD-arknet为主干特征提取网络,YOLO v5 stride-2卷积替换为SPD-Conv模块,分别为主干网络中的5个stride-2卷积层和Neck中的2个stride-2卷积层,可以提高其准确性,同时保持相同级别的参数大小,并向下阶段输出3个不同尺度的特征层;为增强建模通道之间的相互依赖性,自适应地重新校准通道特征响应,引入SE机制提升特征提取能力;为减少模型计算量,提高模型速度,引入SPPF。经测试,改进后YOLO v5网络检测性能最佳,mAP为95.7%,相比YOLO v3、YOLO v4、SSD和YOLO v5网络分别提高4.7、8.8、19.0、3.5个百分点。改进后模型相比改进前对不同温度梯度下的作物病害检测也有提高,5个梯度mAP分别为91.0%、91.6%、90.4%、92.6%和94.0%,分别高于改进前3.6、1.5、7.2、0.6、0.9个百分点。改进YOLO v5网络内存占用量为13.755 MB,低于改进前基础模型3.687 MB。结果表明,改进YOLO v5可以准确快速地实现病害早期检测。

     

    Abstract: To achieve early detection of crop diseases, a crop disease early detection model was proposed based on infrared thermal imaging and improved YOLO v5. The CSPD-arknet was used as the main feature extraction network, and the YOLO v5 stride-2 convolution was replaced by the SPD-Conv module, which were respectively the five stride-2 convolution layers in the main network and the two stride-2 convolution layers in the Neck. This can improve its accuracy while maintaining the same level of parameter size and outputting three different scales of feature layers in the downstream stage. In order to enhance the interdependence between modeling channels, channel feature responses were adaptively recalibrated and SE mechanism was introduced to enhance feature extraction ability. In order to reduce model calculation and improve model speed, SPPF was introduced. After testing, the improved YOLO v5 algorithm had the best detection performance with an mAP of 95.7%, which was respectively 4.7 percentage points, 8.8 percentage points, 19.0 percentage points, and 3.5 percentage points higher than that of YOLO v3, YOLO v4, SSD, and YOLO v5 networks. Compared with the improved network before improvement, it also improved the detection of crop diseases under different temperature gradients. The mAP of five gradients were 91.0%, 91.6%, 90.4%, 92.6%, and 94.0%, which were higher than those before improvement by 3.6 percentage points, 1.5 percentage points, 7.2 percentage points, 0.6 percentage points, and 0.9 percentage points, respectively. The size of the improved YOLO v5 model was 13.755 MB, which was lower than 3.687 MB of the basic network before the improvement. The results showed that improving YOLO v5 can accurately and quickly detect early diseases, which can provide certain technical support for the development of early disease detection instruments.

     

/

返回文章
返回