【计算机科学与应用】YOLO-Apple:一种用于苹果幼果检测的改进型目标检测方法

导读:

针对自然果园场景中,苹果幼果因通体青绿色、与叶片色调相近,且本身尺度偏小、分布密集又易被枝叶遮挡,检测时易出现漏检与误检的问题,文章提出了一种面向苹果幼果检测的改进型目标检测模型YOLO-Apple。该方法以YOLOv11n为基线,在主干C2PSA模块引入轻量特征增强单元Mona (C2PSA_Mona)以强化复杂背景下的特征映射能力,改造C3k2模块并在其增强块嵌入EMA注意力(C3k2_EMA),提升特征表达与选择能力。实验采用来自新疆阿克苏市与石河子市等自然场景采集的5558张苹果幼果图像数据集,按7:2:1划分为训练、验证与测试集。结果表明,YOLO-Apple在测试集上实现P = 91.5%、R = 75.9%、F1 = 83.0、mAP@0.5 = 84.4%的检测性能,在6.5 GFLOPs计算量与2.6 M参数量的轻量级开销下优于多款主流对比模型;消融实验验证了C2PSA_Mona与C3k2_EMA模块的改进有效性,为自然果园苹果幼果检测提供了新方法。

作者信息:

韩昕辰, 安一文*, 武红欣:塔里木大学信息工程学院,新疆 阿拉尔;塔里木绿洲农业教育部重点实验室,新疆 阿拉尔

论文详情

在本研究中,数据集图像来自 AI Studio 的苹果幼果目标识别数据集。

YOLOv11n被选定为本研究的基线模型。通过对 C3k2 模块 和 C2PSA 模块进行针对性改进,构建得到 YOLO-Apple 检测模型,其结构图如图 2 所示。

C2PSA_Mona 的结构图如图 3 所示。

EMA 的结构如图 4 所示。

为选择出最佳模型,本研究使用多种评估指标用于评估模型的性能和计算效率,包括精确率 (Precision, P)、召回率(Recall, R)、mAP@0.5、F1 分数(F1-score)、计算量(GFLOPs)和参数量(Parameters) 等。

所有实验均在同一台设备上进行,该设备配备 AMD5500 处理器,NVIDIA GeForce RTX 5080 显卡, 该显卡具有 16 GB 显存,所使用系统为 Ubuntu 22.04,内存为 48 GB。在实验环境上,使用 Python 3.11.5、 CUDA12.8 和 PyTorch 2.9.1 作为框架。在训练设置上,图片大小为 640 × 640,训练轮次为 200 轮,使用 SGD 作为优化器,批量大小设置为 8。

为验证所提出模块对苹果幼果检测性能的贡献,本文以基线模型(YOLOv11n)为对照,逐步引入各模 块进行消融实验,结果如表 1 所示。

为验证 YOLO-Apple 模型在苹果幼果检测中的性能优势,本研究选取了多种主流的目标检测算法作 为对比实验,包括 YOLOv8n、YOLOv9-tiny、YOLOv10n、YOLOv12n和 YOLOv13n。实验结果如表 2 所示。

本研究针对苹果幼果小目标、遮挡与密集分布等检测难点,提出检测模型 YOLO-Apple。在实验数据 集上,YOLO-Apple 在保持较低复杂度(6.5 GFLOPs,2.6 M 参数)的前提下取得最佳综合性能:P = 91.5%、R = 75.9%、F1 = 83.0、mAP@0.5 = 84.4%。与多种主流轻量模型对比,YOLO-Apple 的 mAP@0.5 相比最 优基线提升约 2.0%,召回率也达到最高,说明其对幼果漏检抑制更有效。消融实验进一步表明,逐步引 入 C2PSA_Mona 与 C3k2_EMA 可稳定提升检测效果,最终在几乎不增加参数量的情况下实现精度与召 回的同步改善。

基金项目:

塔里木大学校长基金胡杨英才项目(TDZKSS202341)

原文链接:

https://doi.org/10.12677/csa.2026.165203

相关推荐
微财经观圈7 小时前
AI 3D生成角色怎样套用预设动作?自动绑骨与动作测试步骤
人工智能·3d
是翎7 小时前
图解大语言模型部署
人工智能·驱动开发·深度学习·开源协议·imagen
joinwell527 小时前
写了“始终审批”,AI 为什么还是执行了?
人工智能
腾渊信息科技公司7 小时前
腾渊科技重磅出品——智能体交易信任架构实战:从KYA到可追溯审计的完整方案
人工智能·科技·aiagent·腾渊科技·腾渊信息科技
知几蜗牛8 小时前
AI写歌有了工程图:YuE2把旋律和和弦放回可编辑层
人工智能
CIO_Alliance8 小时前
AI微调系列(2)| QLoRA:一张卡微调700亿参数,显存是怎么省下来的
大数据·人工智能·深度学习·ai·企业ai转型·企业cio联盟
段一凡-华北理工大学8 小时前
高炉智能布料技术与炉料分布优化~系列文章14:布料制度与送风制度的协同匹配
人工智能·高炉布料·高炉料面·布料制度·送风制度
AI吃大瓜8 小时前
人脸检测和行人检测4:Android实现YOLOv8 YOLO11 YOLO26人脸检测和人体检测(含源码,可实时检测)
android·yolo·人脸检测·人体检测·行人检测·yolo26
thesky1234568 小时前
评测驱动开发(EDD):让 AI 应用“说得清好坏“的方法论
人工智能·ai·大模型
AI备案指南-满满8 小时前
iPhone 18 的 Siri 跟以前到底有什么不一样?
人工智能·ai·生成式ai·大模型备案