YOLO-C: An Efficient and Robust Detection Algorithm for Mature Long Staple Cotton Targets with High-Resolution RGB Images

Abstract

Under complex field conditions, robust and efficient boll detection at maturity is an important tool for pre-harvest strategy and yield prediction. To achieve automatic detection and counting of long-staple cotton in a natural environment, this paper proposes an improved algorithm incorporating deformable convolution and attention mechanism, called YOLO-C, based on YOLOv7: (1) To capture more detailed and localized features in the image, part of the 3 × 3 convolution in the ELAN layer of the backbone is replaced by deformable convolution to improve the expressiveness and accuracy of the model. (2) To suppress irrelevant information, three SENet modules are introduced after the backbone to improve the ability of feature maps to express information, and CBAM and CA are introduced for comparison experiments. (3) A WIoU loss function based on a dynamic non-monotonic focusing mechanism is established to reduce the harmful gradients generated by low-quality examples on the original loss function and improve the model performance. During the model evaluation, the model is compared with other YOLO series and mainstream detection algorithms, and the model mAP@0.5 achieves 97.19%, which is 1.6% better than the YOLOv7 algorithm. In the model testing session, the root mean square error and coefficient of determination (R2) of YOLO-C are 1.88 and 0.96, respectively, indicating that YOLO-C has higher robustness and reliability for boll target detection in complex environments and can provide an effective method for yield prediction of long-staple cotton at maturity.