基于改进ATSS模型的水稻害虫检测

    Rice Pest Detection Based on Improved ATSS Model

    • 摘要:
      目的 针对水稻害虫检测中数据匮乏、精度低、实时性差等问题,构建专用数据集并提出高效检测方法。
      方法 基于诱虫灯采集设备构建Pest5数据集,在ATSS框架上提出改进模型PestDet:采用组合式数据增强策略和锚框优化提升样本多样性与目标匹配能力;以GHM-C和DIoU分别作为分类与回归损失,增强鲁棒性与定位精度;引入膨胀卷积重构特征金字塔,提升多尺度特征感知能力;简化检测头结构并嵌入坐标注意力机制(CA),加快推理速度并强化关键信息提取。
      结果 PestDet在Pest5上mAP达92.0%,FPS为40.2,较原始ATSS分别提升7.0个百分点和8.0,性能优于主流模型。
      结论 PestDet兼具高精度与高效率,可有效识别复杂背景下的水稻害虫,为智能监测与精准防控提供技术支撑。

       

      Abstract:
      Objective To address issues including data scarcity, low accuracy, and poor real-time performance in rice pest detection, this study constructed a specialized dataset and proposed an efficient detection method.
      Method The Pest5 dataset was built based on insect-attracting light traps. Within the ATSS framework, the improved model, PestDet, was proposed. Improvements included: A combined data augmentation strategy and anchor optimization were adopted to enhance sample diversity and target matching capability; GHM-C and DIoU were used as the classification and regression losses, respectively, to improve robustness and localization accuracy; Inflated convolutions was introduced to reconstruct the feature pyramid for enhanced multi-scale feature perception; The detection head architecture was simplified and the Coordinate Attention (CA) mechanism was embedded to accelerate inference and strengthen key information extraction.
      Result PestDet achieved an mAP of 92.0% and FPS of 40.2 on the Pest5 dataset, surpassing the original ATSS by 7.0 percentage points and 8.0 respectively, and outperformed other mainstream models.
      Conclusion PestDet demonstrates high accuracy with high efficiency, enabling effective identification of rice pests in complex backgrounds, and provides technical support for intelligent pest monitoring and precision control.

       

    /

    返回文章
    返回