基于改进型YOLO的复杂环境下番茄果实快速识别方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(31601794)、宁夏回族自治区重点研发计划项目(2018BBF02024)和宁夏回族自治区重点研发计划重大科技项目(2017BY067)


Fast Recognition Method for Tomatoes under Complex Environments Based on Improved YOLO
Author:
Affiliation:

Fund Project:

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

    为实现温室环境下农业采摘机器人对番茄果实的快速、精确识别,提出了一种改进型多尺度YOLO算法(IMSYOLO)。对YOLO网络模型进行筛选和改进,设计了一种含有残差模块的darknet20主干网络,同时融合多尺度检测模块,构建了一种复杂环境下番茄果实快速识别网络模型。该网络模型层数较少,能够提取更多特征信息,且采用多尺度检测结构,同时返回番茄果实的类别和预测框,以此提升番茄果实检测速度和精度。采用自制的番茄数据集对IMSYOLO模型进行测试,并分别对改进前后网络的检测性能以及主干网络层数对特征提取能力的影响进行了对比试验。试验结果表明,IMSYOLO模型对番茄图像的检测精度为97.13%,准确率为96.36%,召回率为96.03%,交并比为83.32%,检测时间为7.719ms;对比YOLO v2和YOLO v3等网络模型,IMSYOLO模型可以同时满足番茄果实检测的精度和速度要求。最后,通过番茄温室大棚采摘试验验证了本文模型的可行性和准确性。

    Abstract:

    In order to implement the fast and accurate recognition of tomatoes for agricultural harvesting robots under greenhouse environments, an improved multiscale YOLO detection algorithm named IMSYOLO was presented. A new backbone network structure, which was named darknet20, with one residual block was designed based on a series of the previous YOLO algorithms, and a multiscale detection structure was utilized simultaneously for the detection algorithm. Therefore, a new kind of neural network model was formed for the fast recognition of tomatoes under complex environments. Due to some features of the method such as the fewer layers required, the larger amount of information extracted, and by using the multiscale structure to return both the detection categories and the bounding boxes, the detection speed and accuracy were improved. IMSYOLO model was tested on our own tomato dataset, and the detection performance of the network before and after the improvement as well as the influence of the variation of the backbone network layers on the feature extraction capacity were analyzed respectively. The test results showed that the proposed method had ideal features with a precision of tomato image detection of 97.13%, an accuracy of 96.36%, a recall rate of 96.03%, an intersection over union (IOU) of 83.32% and a detection time of 7.719ms. Furthermore, compared with YOLO v2, YOLO v3 and some other neural networks mentioned, IMSYOLO can meet the requirements of both detection accuracy and speed. At last, the feasibility of the proposed algorithm applying to the robots was verified by the harvesting tests of the ripe tomatoes under the greenhouse environments.

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

刘芳,刘玉坤,林森,郭文忠,徐凡,张白.基于改进型YOLO的复杂环境下番茄果实快速识别方法[J].农业机械学报,2020,51(6):229-237. LIU Fang, LIU Yukun, LIN Sen, GUO Wenzhong, XU Fan, ZHANG Bai. Fast Recognition Method for Tomatoes under Complex Environments Based on Improved YOLO[J]. Transactions of the Chinese Society for Agricultural Machinery,2020,51(6):229-237.

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