基于改进YOLO v7的轻量化樱桃番茄成熟度检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

山西省基础研究计划项目(202203021212414、202203021212428)和山西农业大学青年科技创新基金项目(J142102257)


Lightweight Maturity Detection of Cherry Tomato Based on Improved YOLO v7
Author:
Affiliation:

Fund Project:

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

    为在自然环境下自动准确地检测樱桃番茄果实的成熟度,实现樱桃番茄果实自动化采摘,根据成熟期樱桃番茄果实表型特征的变化以及国家标准GH/T 1193—2021制定了5级樱桃番茄果实成熟度级别(绿熟期、转色期、初熟期、中熟期和完熟期),并针对樱桃番茄相邻成熟度特征差异不明显以及果实之间相互遮挡问题,提出一种改进的轻量化YOLO v7模型的樱桃番茄果实成熟度检测方法。该方法将MobileNetV3引入YOLO v7模型中作为骨干特征提取网络,以减少网络的参数量,同时在特征融合网络中加入全局注意力机制(Global attention mechanism,GAM)模块以提高网络的特征表达能力。试验结果表明,改进的YOLO v7模型在测试集下的精确率、召回率和平均精度均值分别为98.6%、98.1%和98.2%,单幅图像平均检测时间为82ms,模型内存占用量为66.5MB。对比Faster R-CNN、YOLO v3、YOLO v5s和YOLO v7模型,平均精度均值分别提升18.7、0.2、0.3、0.1个百分点,模型内存占用量也最少。研究表明改进的YOLO v7模型能够为樱桃番茄果实的自动化采摘提供技术支撑。

    Abstract:

    Automatic and accurate detection of cherry tomato maturity in natural environment is the foundation for achieving automatic cherry tomato picking. According to the changes in phenotypic characteristics of cherry tomato during its mature period and the national standard GH/T 1193—2021, and regarding the lack of significant differences in adjacent maturity characteristics of cherry tomatoes and the problem of mutual occlusion between fruits, a lightweight maturity detection method of cherry tomato with five levels, including green, turning, pink, lightred and red was proposed based on improved YOLO v7 model. In this model, MobileNetV3 was introduced into the original YOLO v7 model as backbone for feature extraction to reduce the parameters of the original model; global attention mechanism (GAM) module was added to the feature fusion network to improve the feature expression ability of the model. The experimental results showed that the precision, recall and mean average precision of the improved model were 98.6%, 98.1% and 98.2%, respectively, the average detection time of a single image was 82ms, and the memory occupied by the model was 66.5MB. Compared with Faster R-CNN, YOLO v3, YOLO v5s and YOLO v7 models, the mean average precision (mAP) was improved by 18.7, 0.2, 0.3 and 0.1 percentage points, respectively. The average accuracy of the improved YOLO v7 model was also improved, and memory usage of the model was optimal. The results showed that the improved YOLO v7 model can provide effective exploration for automated cherry tomato fruit picking.

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

苗荣慧,李志伟,武锦龙.基于改进YOLO v7的轻量化樱桃番茄成熟度检测方法[J].农业机械学报,2023,54(10):225-233. MIAO Ronghui, LI Zhiwei, WU Jinlong. Lightweight Maturity Detection of Cherry Tomato Based on Improved YOLO v7[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(10):225-233.

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