论文检索
期刊
全部知识仓储预印本开放期刊机构
高级检索

基于改进COF-YOLO v8n的油茶果静态与动态检测计数方法OA北大核心CSTPCD

Camellia oleifera Fruit Static and Dynamic Detection Counting Based on Improved COF-YOLO v8n

中文摘要英文摘要

针对自然环境下油茶果存在严重遮挡、近景色、小目标等现象,使用YOLO网络存在检测精度低、漏检现象严重等问题,提出对YOLO v8n网络进行改进.首先使用MPDIOU作为YOLO v8n的损失函数,有效解决因为果实重叠导致的漏检问题;其次调整网络,向其中加入小目标检测层,使网络能够关注小目标油茶以及被树叶遮挡的油茶;最后使用SCConv作为特征提取网络,既能兼顾检测精度又能兼顾检测速度.改进COF-YOLO v8n网络精确率、召回率、平均精度均值分别达到97.7%、97%、99%,比未改进的YOLO v8n分别提高3.2、4.8、2.4个百分点,其中严重遮挡情况下油茶检测精确率、召回率、平均精度均值分别达到95.9%、95%、98.5%,分别比YOLO v8n提高4.0、9.1、4.6个百分点.因此改进后COF-YOLO v8n网络能够明显提高油茶在严重遮挡、近景色、小目标均存在情况下的识别精度,减小油茶的漏检.此外,模型能够实现动、静态输入条件下油茶果计数.动态计数借鉴DeepSORT算法的多目标跟踪思想,将改进后COF-YOLO v8n的识别输出作为DeepSORT的输入,实现油茶果实的追踪计数.所得改进模型具有很好的鲁棒性,且模型简单可以嵌入到边缘设备中,不仅可用于指导自动化采收,还可用于果园产量估计,为果园物流分配提供可靠借鉴.

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.

王金鹏;何萌;甄乾广;周宏平

南京林业大学机械电子工程学院,南京 210037

农业科学

油茶果;机器视觉;COF-YOLO v8n;计数;产量估计

Camellia oleifera fruit;machine vision;COF-YOLO v8n;count;production estimates

《农业机械学报》 2024 (004)

193-203 / 11

国家林业和草原局应急科技项目(202202-3)、江苏省农业科技自主创新基金项目(CX(22)3099)、江苏省现代农机装备与技术推广项目(NJ2021-18)和江苏省重点研发计划项目(BE20211016-2)

10.6041/j.issn.1000-1298.2024.04.019

评论

下载量:0
点击量:0