基于视频跟踪算法的果园猕猴桃产量实时预估
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

中央高校基本科研业务费专项资金项目(2662020LXQD002)


Real-time Production Prediction of Kiwifruit in Orchard Based on Video Tracking Algorithm
Author:
Affiliation:

Fund Project:

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

    对猕猴桃产量的准确预估有利于合理安排后续采摘与运输工序,因此开发智能化的产量实时预估工具非常重要。针对大棚培育的猕猴桃矮化密植、分布范围广等特点,本研究利用果园履带小车采集视频,结合人工标注,建立猕猴桃检测和跟踪的数据集。考虑到自制数据集中猕猴桃占比小及密集分布的特点,本文提出使用YOLO v7模型加上Soft-NMS来检测每一帧图像内的猕猴桃。在卡尔曼滤波器预测的结果上,引入VGG16网络对猕猴桃进行特征提取,并结合匈牙利算法完成帧间目标的匹配。最后采用基于YOLO v7+DeepSort跟踪算法的ID计数方法对猕猴桃进行产量估计。实验结果表明,改进的YOLO v7模型在猕猴桃检测数据集上表现良好,检测的F1值为90.09%。猕猴桃跟踪数据集中使用的跟踪算法平均准确率为89.87%,每个目标正确匹配的精确率为82.34%,大型视频跟踪速度为20.19f/s。在环境影响较小的条件下,ID计数准确率为97.49%。该方法可为猕猴桃果园智能化管理中的估产、采收规划等提供技术支撑。

    Abstract:

    The use of machine vision to quickly and accurately estimate fruit yield is of great significance for the development of smart agriculture. In view of the characteristics of dwarf dense planting and wide distribution of kiwifruit cultivated in greenhouses, orchard crawler trolleys were used to shoot and obtain videos of kiwifruit orchards, and a dataset of kiwifruit detection and tracking was established combined with artificial labeling. Considering the small proportion and dense distribution of kiwifruit in the self-made dataset, the YOLO v7 model and Soft-NMS were proposed to detect kiwifruit in each frame. Based on the prediction results of the Kalman filter, the VGG16 network was introduced to extract the features of kiwifruit, and the Hungarian algorithm was used to complete the target matching of the before and after frames. Finally, the ID counting method based on YOLO v7+DeepSort tracking algorithm was used to realize kiwifruit yield estimation. The experimental results showed that the improved YOLO v7 model performed well on the kiwifruit detection dataset, with an F1 score of 90.09%. The average accuracy of the adopted tracking algorithm on the kiwifruit tracking dataset was 89.87%, the precision of each target can be correctly matched was 82.34% and a large video tracking speed of 20.19f/s. Under the condition of low environmental impact, the ID counting accuracy was 97.49%. This method can provide technical support for yield estimation and harvest planning in the intelligent management of kiwifruit orchards.

    参考文献
    相似文献
    引证文献
引用本文

郭明月,刘雅晨,李伟夫,陈洪,李善军,陈耀晖.基于视频跟踪算法的果园猕猴桃产量实时预估[J].农业机械学报,2023,54(6):178-185. GUO Mingyue, LIU Yachen, LI Weifu, CHEN Hong, LI Shanjun, CHEN Yaohui. Real-time Production Prediction of Kiwifruit in Orchard Based on Video Tracking Algorithm[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(6):178-185.

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