A Detection Model for Cucumber Root-Knot Nematodes Based on Modified YOLOv5-CMS

Abstract

The development of resistant cucumber varieties is of a great importance for reducing the production loss caused by root-knot nematodes. After cucumber plants are infected with root-knot nematodes, their roots will swell into spherical bumps. Rapid and accurate detection of the infected sites and assessment of the disease severity play a key role in selecting resistant cucumber varieties. Because the locations and sizes of the spherical bumps formed after different degrees of infection are random, the currently available detection and counting methods based on manual operation are extremely time-consuming and labor-intensive, and are prone to human error. In response to these problems, this paper proposes a cucumber root-knot nematode detection model based on the modified YOLOv5s model (i.e., YOLOv5-CMS) in order to support the breeding of resistant cucumber varieties. In the proposed model, the dual attention module (CBAM-CA) was adopted to enhance the model’s ability of extracting key features, the K-means++ clustering algorithm was applied to optimize the selection of the initial cluster center, which effectively improved the model’s performance, and a novel bounding box regression loss function (SIoU) was used to fuse the direction information between the ground-truth box and the predicted box so as to improve the detection precision. The experiment results show that the recall (R) and mAP of the YOLOv5s-CMS model were improved by 3% and 3.1%, respectively, compared to the original YOLOv5s model, which means it can achieve a better performance in cucumber root-knot nematode detection. This study provides an effective method for obtaining more intuitive and accurate data sources during the breeding of cucumber varieties resistant to root-knot nematode.