Ye Rong, Shao Guoqi, Yang Ziyi, Sun Yuchen, Gao Quan, Li Tong
College of Food Science and Technology, Yunnan Agricultural University, Kunming 650201, China.
The Key Laboratory for Crop Production and Smart Agriculture of Yunnan Province, Kunming 650201, China.
Plants (Basel). 2024 May 15;13(10):1377. doi: 10.3390/plants13101377.
In response to the challenge of low recognition rates for similar phenotypic symptoms of tea diseases in low-light environments and the difficulty in detecting small lesions, a novel adaptive method for tea disease severity detection is proposed. This method integrates an image enhancement algorithm based on an improved EnlightenGAN network and an enhanced version of YOLO v8. The approach involves first enhancing the EnlightenGAN network through non-paired training on low-light-intensity images of various tea diseases, guiding the generation of high-quality disease images. This step aims to expand the dataset and improve lesion characteristics and texture details in low-light conditions. Subsequently, the YOLO v8 network incorporates ResNet50 as its backbone, integrating channel and spatial attention modules to extract key features from disease feature maps effectively. The introduction of adaptive spatial feature fusion in the Neck part of the YOLOv8 module further enhances detection accuracy, particularly for small disease targets in complex backgrounds. Additionally, the model architecture is optimized by replacing traditional Conv blocks with ODConv blocks and introducing a new ODC2f block to reduce parameters, improve performance, and switch the loss function from CIOU to EIOU for a faster and more accurate recognition of small targets. Experimental results demonstrate that YOLOv8-ASFF achieves a tea disease detection accuracy of 87.47% and a mean average precision (mAP) of 95.26%. These results show a 2.47 percentage point improvement over YOLOv8, and a significant lead of 9.11, 9.55, and 7.08 percentage points over CornerNet, SSD, YOLOv5, and other models, respectively. The ability to swiftly and accurately detect tea diseases can offer robust theoretical support for assessing tea disease severity and managing tea growth. Moreover, its compatibility with edge computing devices and practical application in agriculture further enhance its value.
针对弱光环境下茶树病害相似表型症状识别率低以及小病变检测困难的挑战,提出了一种新的茶树病害严重程度自适应检测方法。该方法集成了基于改进的EnlightenGAN网络的图像增强算法和YOLO v8的增强版本。具体做法是,首先通过对各种茶树病害的低光照强度图像进行非配对训练来增强EnlightenGAN网络,引导生成高质量的病害图像。这一步旨在扩大数据集,并改善弱光条件下的病变特征和纹理细节。随后,YOLO v8网络以ResNet50作为骨干,集成通道和空间注意力模块,以有效地从病害特征图中提取关键特征。在YOLOv8模块的Neck部分引入自适应空间特征融合,进一步提高检测精度,特别是对于复杂背景下的小病害目标。此外,通过用ODConv块替换传统的Conv块并引入新的ODC2f块来优化模型架构,以减少参数、提高性能,并将损失函数从CIOU切换到EIOU,从而更快、更准确地识别小目标。实验结果表明,YOLOv8-ASFF的茶树病害检测准确率达到87.47%,平均精度均值(mAP)为95.26%。这些结果表明,与YOLOv8相比提高了2.47个百分点,分别比CornerNet、SSD、YOLOv5等模型显著领先9.11、9.55和7.08个百分点。快速准确地检测茶树病害的能力可为评估茶树病害严重程度和管理茶树生长提供有力的理论支持。此外,它与边缘计算设备的兼容性以及在农业中的实际应用进一步提升了其价值。