基于改进COF-YOLO v8n的油茶果静态与动态检测计数方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

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


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

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对自然环境下油茶果存在严重遮挡、近景色、小目标等现象,使用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的输入,实现油茶果实的追踪计数。所得改进模型具有很好的鲁棒性,且模型简单可以嵌入到边缘设备中,不仅可用于指导自动化采收,还可用于果园产量估计,为果园物流分配提供可靠借鉴。

    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的油茶果静态与动态检测计数方法[J].农业机械学报,2024,55(4):193-203. WANG Jinpeng, HE Meng, ZHEN Qianguang, ZHOU Hongping. Camellia oleifera Fruit Static and Dynamic Detection Counting Based on Improved COF-YOLO v8n[J]. Transactions of the Chinese Society for Agricultural Machinery,2024,55(4):193-203.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-08-11
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-10
  • 出版日期: