华中农业大学学报

作者：陈仁凡，谢知，林晨

单位：1.福建农林大学机电工程学院，福州 350002;2.福建省农业信息感知技术重点实验室，福州 350002

关键词：草莓;成熟度;目标检测;YOLOv5s;Shuffle_Block;精准识别

基金：福建省自然科学基金项目（2019J01403）

为解决温室环境下草莓果实快速准确识别问题，提出一种基于改进YOLOv5s的草莓成熟度检测方法。在主干中引入Shuffle_Block作为特征提取网络，从而实现模型轻量化。同时，在颈部结构中使用全维度动态卷积模块（omni-dimensional dynamic convolution， ODConv），以提高网络对草莓目标的信息挖掘能力，降低计算量，并进一步实现轻量化。结果显示，改进后的YOLO-ODM（YOLO with ODConv module）模型的平均精度均值达97.4%，模型体积是7.79 Mb，在GPU上的单张平均检测时间仅11 ms，浮点运算量为6.9×109。与原网络相比，轻量化的YOLO-ODM方法在提高检测精度的同时，模型大小缩减43%，浮点运算量降低52%。以上结果表明，该轻量化方法可快速准确地对温室环境下草莓果实的成熟度进行检测，实现草莓的生长状态监测。

来源：2023年第4期

《华中农业大学学报》期刊编辑部

查看华中农业大学学报杂志2023年第4期