针对小麦病虫害分割中的像素不平衡问题,提出新数据增强方法。
Augment to Segment: Tackling Pixel-Level Imbalance in Wheat Disease and Pest Segmentation
- 从标注图像中提取稀有虫害区域,随机变换后贴入新位置。
- 引入随机投影滤波,使贴入区域更自然,提升特征融合效果。
- 显著提升虫害类别的分割精度,适合农业视觉任务研究者。
小麦叶片病害与虫害的精准分割对作物管理至关重要,但虫害通常仅占标注像素的一小部分,造成极端的像素级不平衡,导致模型过拟合常见类别而忽略稀有类别,影响整体性能。本文提出一种随机投影复制粘贴(RPCP)数据增强技术:从训练图像中提取稀有虫害区域,施加随机几何变换以模拟变化,并将其贴入合适区域,避免与病斑或已有损伤重叠;同时使用随机投影滤波器对贴入区域进行处理,优化局部特征,确保与背景自然融合。实验表明,该方法显著提升了虫害类别的分割性能,同时保持甚至小幅提高其他类别的准确率。结果验证了针对性增强在缓解极端像素不平衡方面的有效性,为农业图像分割提供了简单而高效的解决方案。
原文摘要 · Abstract (English)
Accurate segmentation of foliar diseases and insect damage in wheat is crucial for effective crop management and disease control. However, the insect damage typically occupies only a tiny fraction of annotated pixels. This extreme pixel-level imbalance poses a significant challenge to the segmentation performance, which can result in overfitting to common classes and insufficient learning of rare classes, thereby impairing overall performance. In this paper, we propose a Random Projected Copy-and-Paste (RPCP) augmentation technique to address the pixel imbalance problem. Specifically, we extract rare insect-damage patches from annotated training images and apply random geometric transformations to simulate variations. The transformed patches are then pasted in appropriate regions while avoiding overlaps with lesions or existing damaged regions. In addition, we apply a random projection filter to the pasted regions, refining local features and ensuring a natural blend with the new background. Experiments show that our method substantially improves segmentation performance on the insect damage class, while maintaining or even slightly enhancing accuracy on other categories. Our results highlight the effectiveness of targeted augmentation in mitigating extreme pixel imbalance, offering a straightforward yet effective solution for agricultural segmentation problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。