基于SAB-YOLO模型的生菜生长期识别方法
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

上海市科委科技创新行动计划项目(23N21900400)


Lettuce Growth Period Recognition Method Based on SAB-YOLO Model
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对作物生长期难以自动化识别、传统机器学习识别方法的精度有限、不同阶段尺度差异大导致识别准确度较低的问题,本文提出一种生长期识别模型SAB-YOLO(Self attention based-YOLO)。将YOLO v5特征提取网络替换为基于自注意力机制的Swin Transformer网络,增强模型对全局特征的捕捉能力;并将Neck网络改进为跨层连接更为密集的AFPN结构,改善多尺度特征融合效果;提出了将卷积和自注意力机制结合的CTF (Convolutional transformer fusion)模块,并应用在检测头位置以增强全局特征;最后将损失函数改为Inner-SIoU。试验结果表明,改进模型在测试集上精确率达到88.5%、mAP_0.5为92.1%,提升了生菜图像生长期识别精度。研究结果为作物生长期识别提供了新的技术方案,对精准农业发展具有实践价值。

    Abstract:

    In container-based vertical agricultural production systems, supplementary lighting is a key technical approach for regulating crop growth, optimizing resource utilization, and improving production efficiency. However, most existing lighting control strategies rely on fixed time cycles or empirical parameters, lacking effective perception of crop growth stages, which limits their adaptability and precision under dynamic growth conditions. To address these challenges, a lettuce growth stage recognition model named self attention based-YOLO (SAB-YOLO) was proposed to realize accurate and automated identification of crop growth periods in complex visual environments. The proposed model was developed by introducing multiple structural improvements to the YOLO v5 framework. Firstly, the conventional convolutional backbone was replaced with a Swin Transformer network based on self-attention mechanisms, which enhanced the ability of the model to capture long-range dependencies and global semantic information. Secondly, an asymptotic feature pyramid network (AFPN) with denser cross-layer connections was adopted in the Neck to strengthen multi-scale feature fusion and improve robustness to large scale variations among different growth stages. Furthermore, a convolution transformer fusion (CTF) module that integrated convolutional operations with self-attention was designed and embedded into the detection head to further enhance global contextual representation. In addition, the Inner-SIoU loss function was employed to improve bounding box regression accuracy and accelerate model convergence. Experimental results on a mixed dataset collected from open-source platforms and a container-based plant factory showed that the proposed model achieved a precision of 88.5% and an mAP_0.5 of 92.1%, outperforming the baseline YOLO v5 model. Furthermore, an intelligent supplementary lighting system based on growth stage recognition was designed and validated, demonstrating the practical applicability of the proposed method in precision agriculture.

    参考文献
    相似文献
    引证文献
引用本文

林开颜,王先浪,牛程远,吴军辉,陈杰,杨学军.基于SAB-YOLO模型的生菜生长期识别方法[J].农业机械学报,2026,57(6):290-299. LIN Kaiyan, WANG Xianlang, NIU Chengyuan, WU Junhui, CHEN Jie, YANG Xuejun. Lettuce Growth Period Recognition Method Based on SAB-YOLO Model[J]. Transactions of the Chinese Society for Agricultural Machinery,2026,57(6):290-299.

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-08-18
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-04-15
  • 出版日期:
文章二维码