北方农业学报 ›› 2023, Vol. 51 ›› Issue (5): 114-122.doi: 10.12190/j.issn.2096-1197.2023.05.12

• 农业信息技术 • 上一篇    下一篇

温室樱桃番茄果实深度学习识别模型研究

沈超1,2, 夏秀波1,3, 杨玮2, 张焕春1,3, 李民赞2   

  1. 1.山东省烟台市农业科学研究院,山东 烟台 265500
    2.中国农业大学智慧农业系统集成研究教育部重点实验室,北京 100083
    3.烟台市智慧农业研究中心,山东 烟台 265500
  • 收稿日期:2023-08-02 出版日期:2023-10-20 发布日期:2024-01-04
  • 通讯作者: 杨 玮(1981—),女,副教授,博士,主要从事精细农业系统集成方面的研究工作。夏秀波(1980—),男,高级农艺师,硕士,主要从事蔬菜育种与智能化栽培方面的研究工作。
  • 作者简介:沈超(1996—),男,硕士研究生,研究方向为智慧农业。
  • 基金资助:
    山东省烟台市科技计划项目(2020XCZX047);2023年烟台市设施番茄育种攻关团队项目

Research on deep learning recognition model of greenhouse cherry tomato fruit

SHEN Chao1,2, XIA Xiubo1,3, YANG Wei2, ZHANG Huanchun1,3, LI Minzan2   

  1. 1. Yantai Academy of Agricultural Sciences of Shandong,Yantai 265500,China
    2. Key Laboratory of Smart Agriculture Systems,Ministry of Education,China Agricultural University,Beijing 100083,China
    3. Yantai Smart Agriculture Research Center,Yantai 265500,China
  • Received:2023-08-02 Online:2023-10-20 Published:2024-01-04

摘要:

【目的】选取樱桃番茄果实深度学习识别模型,实现对温室樱桃番茄果实快速、准确的检测。【方法】采集樱桃番茄果实样本,标注数据,构建数据集,对YOLOv4和YOLOv4-Tiny算法模型进行训练,并对训练得到的模型进行精确率、召回率、平均准确率、综合评价指标F1值的分析。【结果】在训练好的YOLOv4和YOLOv4-Tiny算法模型上,樱桃番茄果实识别预测准确率分别为100%和96.84%、召回率分别为91.12%和90.53%、平均精确率分别为96.32%和94.18%、综合评价指标F1值分别为0.95和0.94。【结论】YOLOv4算法模型明显优于YOLOv4-Tiny,能够实现对樱桃番茄果实目标的准确检测。

关键词: 深度学习, 樱桃番茄, 果实, 识别, YOLOv4, YOLOv4-Tiny

Abstract:

【Objective】Select deep learning cherry tomato fruit recognition models to achieve rapid and accurate detection of greenhouse cherry tomato fruits.【Methods】The training data set was constructed by collecting the cherry tomato fruit samples and annotating the data. YOLOv4 and YOLO v4-Tiny algorithms were used for model training. The precision,recall,mean average precision and comprehensive evaluation index F1 value of the trained models were analyzed.【Results】The precision of cherry tomato fruit recognition prediction was 100% and 96.84% for the trained YOLOv4 and YOLOv4-Tiny models,the recall rates were 91.12% and 90.53%,average precision was 96.32% and 94.18%,and the comprehensive evaluation index F1 values were 0.95 and 0.94,respectively.【Conclusion】YOLOv4 algorithm model was obviously better than YOLOv4-Tiny,could achieve accurate cherry tomato fruit recognition.

Key words: Deep learning, Cherry tomato, Fruit, Recognition, YOLOv4, YOLOv4-Tiny

中图分类号: 

  • TP391.41