通过基于生成式人工智能(GenAI)的图像增强提升杂草检测性能
Enhancing weed detection performance by means of GenAI-based image augmentation
AI总结:
针对精准杂草管理中高质量标注训练数据稀缺的问题,本文提出基于Stable Diffusion模型的生成式AI图像增强技术,提升杂草检测模型训练数据集的数量和质量,实验表明该技术能显著提高YOLO模型的检测性能。
摘要:
精准杂草管理对维持作物生产力和生态平衡至关重要。传统除草剂施用面临经济和环境挑战,凸显了对深度学习驱动的智能杂草控制系统的需求。这些系统需要大量高质量训练数据。然而,标注良好的训练数据稀缺这一现实,通常通过数据增强生成更多数据来解决。尽管如此,随机翻转、颜色变化和模糊等传统增强技术缺乏足够的保真度和多样性。本文研究了一种基于生成式人工智能的增强技术,该技术使用Stable Diffusion模型生成多样化的合成图像,以提升杂草检测模型训练数据集的数量和质量。此外,本文探索了这些合成图像对实时检测系统性能的影响,因此聚焦于适用于边缘设备的紧凑型CNN模型(如YOLO nano)。实验结果表明,使用生成式人工智能增强数据集训练的YOLO模型在平均精度均值(mAP50和mAP50-95)得分上有显著提升,证明了合成数据在增强模型鲁棒性和准确性方面的潜力。
英文摘要:
Precise weed management is essential for sustaining crop productivity and ecological balance. Traditional herbicide applications face economic and environmental challenges, emphasizing the need for intelligent weed control systems powered by deep learning. These systems require vast amounts of high-quality training data. The reality of scarcity of well-annotated training data, however, is often addressed through generating more data using data augmentation. Nevertheless, conventional augmentation techniques such as random flipping, color changes, and blurring lack sufficient fidelity and diversity. This paper investigates a generative AI-based augmentation technique that uses the Stable Diffusion model to produce diverse synthetic images that improve the quantity and quality of training datasets for weed detection models. Moreover, this paper explores the impact of these synthetic images on the performance of real-time detection systems, thus focusing on compact CNN-based models such as YOLO nano for edge devices. The experimental results show substantial improvements in mean Average Precision (mAP50 and mAP50-95) scores for YOLO models trained with generative AI-augmented datasets, demonstrating the promising potential of synthetic data to enhance model robustness and accuracy.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。