基于改进YOLOX的自然环境下核桃识别算法研究
钟正扬1
云利军1
杨璇玺2
陈载清1
1.云南师范大学 信息学院,云南 昆明 650500;云南省教育厅计算机视觉与智能控制技术工程研究中心,云南 昆明 6505002.云南省林业调查规划院 生态分院卫星林业应用中心,云南 昆明 650500
摘要：针对现有目标检测算法对自然环境下核桃识别存在漏检、误检等问题,提出了一种基于Swin Transformer多层特征融合改进的YOLOX-S核桃识别算法.首先,在主干特征提取网络中引入基于Swin Transformer的多层特征融合模块,借助Swin Transformer的多头注意力机制对小目标的特征信息进行提取并与特征图进行融合,可以有效解决因网络层数加深导致的高层特征图中小目标特征信息丢失问题;其次,为了提高算法的检测精度,引入更高效的Repblock模块对原网络中的CSP模块进行替换;最后,为了提高下采样效果,使用更为优秀的Transition Block模块作为主干特征提取网络的下采样模块.结果表明,改进后的YOLOX-S模型在采集的自然环境下核桃数据集上平均精度AP50达到96.72%,分别比Faster-RCNN、YOLOv5-S、YOLOX-S算法提高7.36、1.38、0.62百分点,检测速度达到46 f/s,模型参数大小为20.55 M.改进后的YOLOX-S算法具有更好的精度,改善了漏检和误检问题,对自然环境下的核桃有更好的识别效果.
关键词：核桃识别Swin Transformer多层特征融合模块YOLOX-S深度学习
分类号：S664.1(果树园艺)S126(农业物理学)
资助基金：云南省教育厅科学研究基金项目(2023Y0533)
论文发表日期：2024-01-15
在线出版日期：2026-09-12（本平台首次上网日期，不代表文献的发表时间）
页数：10( 152-161 )
英文信息