基于卷积神经网络的草莓识别方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(51769010、51979133、51469010)


Identification Method of Strawberry Based on Convolutional Neural Network
Author:
Affiliation:

Fund Project:

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

    针对目前草莓识别定位大多在简单环境下进行、识别效率较低的问题,提出利用改进的YOLOv3识别方法在复杂环境中对草莓进行连续识别检测。通过训练大量的草莓图像数据集,得到最优权值模型,其测试集的精度均值(MAP)达到87.51%;成熟草莓的识别准确率为97.14%,召回率为94.46%;未成熟草莓的识别准确率为96.51%,召回率为93.61%。在模型测试阶段,针对夜晚环境下草莓图像模糊的问题,采用伽马变换得到的增强图像较原图识别正确率有显著提升。以调和平均值(F)作为综合评价指标,对比多种识别方法在不同果实数量、不同时间段及视频测试下的实际检测结果,结果表明,YOLOv3算法F值最高,每帧图像的平均检测时间为34.99ms,视频的平均检测速率为58.1f/s,模型的识别正确率及速率均优于其他算法,满足实时性要求。同时,该方法在果实遮挡、重叠、密集等复杂环境下具有良好的鲁棒性。

    Abstract:

    Aiming to solve the problems that strawberry identification and localization were mostly carried out in a simple environment and the identification efficiency was low, the continuous identification and detection of strawberry in a complex environment was studied, and an improved YOLOv3 identification method was proposed. By training a large number of strawberry image data sets, the optimal weight model was obtained. The mean average precision (MAP) of the test set reached 87.51%, among which the average accuracy and the recall rate of mature strawberries was 97.14% and 94.46%, respectively, and that of immature strawberries was 96.51% and 93.61%. In the model test stage, aiming at the problem of strawberry image blurring in the night environment, the recognition accuracy of the original image was significantly improved by using Gamma transform image enhancement. The harmonic mean value (F value) was used as the comprehensive evaluation index, and the actual test results of various identification methods under different fruit numbers, time periods and video tests were compared. The results showed that the improved YOLOv3 algorithm had the highest F value, the average detection time of the picture was 34.99ms, and the average detection frame rate of the video was 58.1f/s, indicating that the recognition accuracy and rate of the model were better than that of other algorithms, and it had good robustness in complex environments such as fruit occlusion, overlap and density. This study can provide theoretical basis for continuous operation of strawberry picking robot under actual working conditions.

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

刘小刚,范诚,李加念,高燕俐,章宇阳,杨启良.基于卷积神经网络的草莓识别方法[J].农业机械学报,2020,51(2):237-244. LIU Xiaogang, FAN Cheng, LI Jianian, GAO Yanli, ZHANG Yuyang, YANG Qiliang. Identification Method of Strawberry Based on Convolutional Neural Network[J]. Transactions of the Chinese Society for Agricultural Machinery,2020,51(2):237-244.

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