基于改进YOLOv5n的花生荚果实时检测方法  

Real‑time peanut pods detection method based on improved YOLOv5n

作　　者：吴阳华 王建楠 刘敏基 游兆延 谢焕雄 杜元杰 Wu Yanghua;Wang Jiannan;Liu Minji;You Zhaoyan;Xie Huanxiong;Du Yuanjie(Nanjing Institute of Agricultural Mechanization,Ministry of Agriculture and Rural Affairs,Nanjing,210014,China;Chinese Academy of Agricultural Sciences,Beijing,100081,China)

机构地区：[1]农业农村部南京农业机械化研究所,南京市210014 [2]中国农业科学院,北京市100081

出　　处：《中国农机化学报》2025年第2期230-236,258,共8页Journal of Chinese Agricultural Mechanization

基　　金：国家重点研发项目子课题(2023YFD2001005—3);国家花生产业技术体系产后加工机械化岗位(CARS—13—产后加工机械化)。

摘　　要：花生荚果分级是花生商品化过程中的重要环节。针对传统花生荚果分级机械精度较低、局限性较大等问题,提出一种基于深度学习的YOLOv5n—SP花生荚果检测算法,对花生荚果按籽仁数量和是否破损进行分级。结合GSConv构建轻量级颈部网络,轻量化的同时实现性能提升;为减少计算冗余,引入Slimming算法进行通道剪枝,在保证性能的前提下进一步降低模型参数量;引入通道智慧蒸馏算法,提高剪枝模型性能。结果表明,改进后的YOLOv5n—SP相较于原模型YOLOv5n,浮点计算量减少58.5%,模型精确率、召回率分别提高2.0%和1.1%,实时检测速度达到84帧/s,提升7.7%。The grading of peanut pods is a crucial step in the commercialization process of peanuts.In view of the low mechanical precision and large limitations of traditional peanut pod grading machines,a YOLOv5n—SP algorithm for peanut pod detection based on deep learning was proposed,which could classify peanut pods according to the number of kernels and whether they were damaged.By incorporating the GSConv,a lightweight neck network was constructed,achieving performance improvement while reducing computational complexity.In order to mitigate computational redundancy,the Slimming algorithm was employed for channel pruning,further reducing the parameter count while maintaining performance.Additionally,the channel wise distillation algorithm was introduced to enhance the performance of pruned model.Experimental results demonstrate that compared to the original YOLOv5n model,the improved YOLOv5n—SP model reduces floating point operations by 58.5%,with a 2.0%and 1.1%increase in model precision and recall rate.The real-time detection speed reaches 84 frame/s,representing a 7.7%improvement.

关 键 词：花生荚果 分级机械 轻量化模型 剪枝算法 智慧蒸馏算法 实时检测 

分 类 号：S226.5[农业科学—农业机械化工程]