Research on Apple Surface Disease Detection Method Based on Improved YOLOv11s.

Abstract

Apple surface diseases are crucial factors affecting the quality and yield of apples. Traditional manual inspection methods suffer from low efficiency and poor real-time performance. To address these issues, this paper proposes an apple surface disease detection method based on an improved YOLOv11s. Firstly, three groups of GAM attention mechanisms are integrated into the neck structure of the YOLOv11s to enhance the efficiency of feature fusion and the capability of semantic information transmission. Secondly, the original convolutional downsampling in the backbone network is replaced with a Haar-based feature downsampling module, enabling the model to retain more high-frequency detail information during the downsampling process. In addition, the WFU module is introduced to realize the dynamic allocation of feature weights, enhancing the model's ability to recognize multi-scale defect features. Finally, the PIOUv2 loss function is adopted to optimize bounding box regression, improving the model's detection performance for tiny defect spots. In addition, various data augmentation methods for small datasets are employed to improve the model training performance and effectively avoid the problem of data overfitting. The experimental results demonstrate that the F1-score of the proposed model is increased by 4.2%, and the mAP@50:95 is boosted by 2.4%. The detection performance outperforms various comparative models, which verifies the effectiveness and superiority of the proposed method.

Plant phenotyping relevance

リンゴ表面の病斑・欠陥を画像から検出する改良YOLO手法の開発と比較検証が研究の中心であり、植物の病害状態を直接推定するフェノタイピング手法に該当する。

Code and data availability

The paper uses a hybrid apple disease image dataset whose public portion is explicitly cited as reference [27] with a public Baidu Netdisk URL (the only allowed URL), matching the paper's apple surface disease detection dataset. The self-built portion and raw data are available only on request, and no author analysis代码