农业机械学报2024,Vol.55Issue(4):193-203,11.DOI:10.6041/j.issn.1000-1298.2024.04.019
基于改进COF-YOLO v8n的油茶果静态与动态检测计数方法
Camellia oleifera Fruit Static and Dynamic Detection Counting Based on Improved COF-YOLO v8n
摘要
Abstract
Aiming at the problems of severe occlusion,close view and small target Camellia oleifera in Camellia oleifera fruit,the original YOLO v8n network was selected to improve the phenomenon of low detection accuracy and serious missed detection phenomenon by using the original YOLO network.MPDIOU was used as the loss function of YOLO v8n to effectively solve the problem of missed detection caused by fruit overlap.Adjusting the network and adding a small target detection layer to it,so that the network can pay attention to small target Camellia oleifera and Camellia oleifera obscured by leaves;SCConv was used to replace the C2f in the original YOLO v8n,so that the network can balance both detection accuracy and detection speed.The P,R and mAP of the improved COF-YOLO v8n network reached 97.7%,97%and 99%respectively,which were 3.2 percentages,4.8 percentages and 2.4 percentages higher than P,R and mAP of the unimproved YOLO v8n,among which the P,R and mAP of Camellia oleifera reached 95.9%,95%and 98.5%under severe occlusion,respectively,which was 4.0 percentages,9.1 percentages and 4.6 percentages higher than that of the original YOLO v8n.The COF-YOLO v8n network can significantly improve the recognition accuracy of Camellia oleifera under the conditions of severe occlusion,close vie,and small targets.In addition,the model can realize the counting of Camellia oleifera under dynamic and static input conditions.Dynamic counting draws on the multi-target tracking idea of DeepSORT algorithm,and took the recognition output of COF-YOLO v8n as the input of DeepSORT to realize the recognition and counting of Camellia oleifera fruits,and used the reduced resolution Camellia oleifera data to simulate the target situation in the field environment and restored the real picking environment.The resulting improved model had good robustness and simple model can be embedded in the robotic arm,which can not only be used to guide future automated harvesting,but also for yield estimation of orchards,providing reliable reference for orchard logistics distribution.关键词
油茶果/机器视觉/COF-YOLO v8n/计数/产量估计Key words
Camellia oleifera fruit/machine vision/COF-YOLO v8n/count/production estimates分类
农业科技引用本文复制引用
王金鹏,何萌,甄乾广,周宏平..基于改进COF-YOLO v8n的油茶果静态与动态检测计数方法[J].农业机械学报,2024,55(4):193-203,11.基金项目
国家林业和草原局应急科技项目(202202-3)、江苏省农业科技自主创新基金项目(CX(22)3099)、江苏省现代农机装备与技术推广项目(NJ2021-18)和江苏省重点研发计划项目(BE20211016-2) (202202-3)