基于深度学习的轻量化草莓成熟度检测模型OA
A lightweight model for strawberry ripeness detection based on deep learning
为实现复杂环境下草莓成熟度的实时检测,提出一种基于YOLOv5s改进的轻量化目标检测模型YOLOv5s-DCC,以期达到精度和轻量化的双重提升.将深度可分离卷积嵌入骨干网络(Backbone)结构中降低计算复杂度,在颈部网络(Neck)部分引入CARAFE上采样增强对细小特征捕获的能力,在检测头(Head)部分融合CBAM注意力机制,通过动态权重提升特征选择性能,构建了兼顾检测精度与轻量化的优化模型.改进后的模型大小仅为13.9 MB,精确率、召回率、平均精度(mAP)分别为93.4%、92.7%、95.3%.相比于原模型YOLOv5s,精确率、召回率、平均精度分别提高1.3、0.9和1.6个百分点,参数量减少0.5×106.相较于YOLOX-s、YOLOv7-Tiny和YOLOv8s等主流轻量化模型,YOLOv5s-DCC综合性能最优,能够满足复杂环境下农业采摘机器人对草莓果实成熟度的实时检测需求.
To achieve real-time detection of strawberry ripeness in complex environments,a lightweight object detection model YO-LOv5s-DCC based on an enhanced YOLOv5s architecture was proposed,aiming to enhance both accuracy and computational efficien-cy.Depthwise separable convolutions were embedded within the backbone structure to reduce computational complexity.The neck sec-tion incorporated CARAFE upsampling to enhance the capture of minute features,while the detection head integrated CBAM attention mechanisms.Dynamic weighting improved feature selection performance,resulting in an optimized model balancing detection accura-cy and lightweight efficiency.The refined model size was only 13.9 MB,achieving precision,recall,and mean average precision(mAP)of 93.4%,92.7%,and 95.3%,respectively.Compared to the original YOLOv5s model,precision,recall,and mAP improved by 1.3,0.9,and 1.6 percentage points,respectively,while reducing parameters by 0.5×106.Compared to mainstream lightweight mod-els such as YOLOX-s,YOLOv7-tiny,and YOLOv8s,YOLOv5s-DCC deliverd the best overall performance.It could meet the real-time strawberry ripeness detection requirements of agricultural harvesting robots in complex environments.
李慧琴;王洋洋;颉世国;王鹏飞;兰明明
河南农业大学机电工程学院,郑州 450002河南农业大学机电工程学院,郑州 450002河南农业大学机电工程学院,郑州 450002河南农业大学机电工程学院,郑州 450002河南农业大学机电工程学院,郑州 450002
信息技术与安全科学
草莓成熟度检测轻量化模型深度可分离卷积CBAM注意力机制CARAFE上采样
strawberry ripeness detectionlightweight modeldepth separable convolutionCBAM attention mechanismCARAFE upsampling
《湖北农业科学》 2026 (4)
7-15,9
河南省高等学校重点科研项目计划(26A460012)
评论