用改进YOLO11检测棉花铃成熟阶段,提升田间识别精度。
COTONET: A custom cotton detection algorithm based on YOLO11 for stage of growth cotton boll detection
- 引入注意力机制与特征重组装,增强难检棉铃识别能力。
- 在标准数据集上达到mAP50 81.1%、mAP50-95 60.6%。
- 模型轻量(7.6M参数),适合边缘设备部署,适配农业机器人。
棉花采收是关键阶段,棉铃的物理操作易导致纤维劣化。为维持高品质,采摘方式需模拟精细的人工抓握。自动化该过程需能识别不同生长阶段棉铃的系统。为此,我们提出COTONET,一种基于改进YOLO11的定制检测算法,融合注意力机制以提升难检实例的检测性能。架构通过非可学习操作中的梯度优化,增强形状与特征提取能力。核心改进包括:用Squeeze-and-Exitation模块替代卷积块,重构主干网络集成注意力机制,以内容感知特征重组装(CARAFE)替代标准上采样;同时在主干中加入简单注意力模块(SimAM),在下向颈部路径采用并行混合注意力机制(PHAM),实现通道、空间和坐标维度的联合注意力。该设计显著提升对复杂棉株生长状态的解析能力。COTONET参数量为7.6M,计算量27.8 GFLOPS,属小中型模型,适用于低资源边缘计算与移动机器人。相比标准YOLO基线,COTONET在测试中实现mAP50 81.1%、mAP50-95 60.6%。
原文摘要 · Abstract (English)
Cotton harvesting is a critical phase where cotton capsules are physically manipulated and can lead to fibre degradation. To maintain the highest quality, harvesting methods must emulate delicate manual grasping, to preserve cotton's intrinsic properties. Automating this process requires systems capable of recognising cotton capsules across various phenological stages. To address this challenge, we propose COTONET, an enhanced custom YOLO11 model tailored with attention mechanisms to improve the detection of difficult instances. The architecture incorporates gradients in non-learnable operations to enhance shape and feature extraction. Key architectural modifications include: the replacement of convolutional blocks with Squeeze-and-Exitation blocks, a redesigned backbone integrating attention mechanisms, and the substitution of standard upsampling operations for Content Aware Reassembly of Features (CARAFE). Additionally, we integrate Simple Attention Modules (SimAM) for primary feature aggregation and Parallel Hybrid Attention Mechanisms (PHAM) for channel-wise, spatial-wise and coordinate-wise attention in the downward neck path. This configuration offers increased flexibility and robustness for interpreting the complexity of cotton crop growth. COTONET aligns with small-to-medium YOLO models utilizing 7.6M parameters and 27.8 GFLOPS, making it suitable for low-resource edge computing and mobile robotics. COTONET outperforms the standard YOLO baselines, achieving a mAP50 of 81.1% and a mAP50-95 of 60.6%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。