针对复杂果园中沙田柚检测难题,提出高效精准的SDE-DET模型。
SDE-DET: A Precision Network for Shatian Pomelo Detection in Complex Orchard Environments
- 采用星形模块与可变形注意力,增强遮挡下特征提取能力。
- 在自建数据集上达到[email protected]:0.95为0.823,优于主流模型。
- 适合智能采摘机器人开发,尤其适用于小目标与多尺度检测场景。
沙田柚检测是其定位、自动化机器人采摘及成熟度分析的关键环节。然而,在复杂果园环境中检测沙田柚面临多尺度、树干与枝叶遮挡、小目标检测等挑战。为此,本研究构建了定制数据集STP-AgriData,提出SDE-DET模型用于沙田柚检测。该模型首先利用星形模块(Star Block)在不增加计算开销的前提下有效获取高维信息;其次,在主干网络中引入可变形注意力机制,提升遮挡条件下的检测能力;最后,集成多种高效多尺度注意力机制,在降低计算负担的同时提取深层视觉表征,增强小目标检测性能。实验对比了SDE-DET与Yolo系列及其他主流检测模型在沙田柚检测中的表现。结果表明,SDE-DET在精确率、召回率、[email protected]、[email protected]:0.95和F1分数上分别达到0.883、0.771、0.838、0.497和0.823,优于现有方法,在STP-AgriData数据集上达到当前最优水平。实验验证了SDE-DET在沙田柚检测中的可靠性,为自动采收机器人的进一步发展奠定基础。
原文摘要 · Abstract (English)
Pomelo detection is an essential process for their localization, automated robotic harvesting, and maturity analysis. However, detecting Shatian pomelo in complex orchard environments poses significant challenges, including multi-scale issues, obstructions from trunks and leaves, small object detection, etc. To address these issues, this study constructs a custom dataset STP-AgriData and proposes the SDE-DET model for Shatian pomelo detection. SDE-DET first utilizes the Star Block to effectively acquire high-dimensional information without increasing the computational overhead. Furthermore, the presented model adopts Deformable Attention in its backbone, to enhance its ability to detect pomelos under occluded conditions. Finally, multiple Efficient Multi-Scale Attention mechanisms are integrated into our model to reduce the computational overhead and extract deep visual representations, thereby improving the capacity for small object detection. In the experiment, we compared SDE-DET with the Yolo series and other mainstream detection models in Shatian pomelo detection. The presented SDE-DET model achieved scores of 0.883, 0.771, 0.838, 0.497, and 0.823 in Precision, Recall, [email protected], [email protected]:0.95 and F1-score, respectively. SDE-DET has achieved state-of-the-art performance on the STP-AgriData dataset. Experiments indicate that the SDE-DET provides a reliable method for Shatian pomelo detection, laying the foundation for the further development of automatic harvest robots.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。