基于改进YOLO v7-tiny的甜椒畸形果识别算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

岭南现代农业科学与技术广东省实验室科研项目(NT2021009)、国家自然科学基金面上项目(32372002)、广东省农业科学院科技人才引进专项资金项目(R2019YJ-YB3003)、广东省农业科学院协同创新中心项目(XT202201)和广东省重点领域研发计划项目(2023B0202090001)


Malformed Sweet Pepper Fruit Identification Algorithm Based on Improved YOLO v7-tiny
Author:
Affiliation:

Fund Project:

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

    甜椒在生长发育过程中容易产生畸形果,机器代替人工对甜椒畸形果识别和摘除一方面可提高甜椒品质和产量,另一方面可解决当前人工成本过高、效率低下等问题。为实现机器人对甜椒果实的识别,提出了一种基于改进YOLO v7-tiny目标检测模型,用于区分正常生长和畸形生长的甜椒果实。将无参数注意力机制(Parameterfree attention module, SimAM)融合到骨干特征提取网络中,增强模型的特征提取和特征整合能力;用Focal-EIOU(Focal and efficient intersection over union)损失替换原损失函数CIOU(Complete intersection over union),加快模型收敛并降低损失值;使用SiLU激活函数代替原网络中的Leaky ReLU,增强模型的非线性特征提取能力。试验结果表明,改进后的模型整体识别精确度、召回率、平均精度均值(Mean average precision, mAP)mAP0.5、mAP0.5-0.95分别为99.1%、97.8%、98.9%、94.5%,与改进前相比,分别提升5.4、4.7、2.4、10.7个百分点,模型内存占用量为 10.6MB,单幅图像检测时间为4.2ms。与YOLO v7、Scaled-YOLO v4、YOLOR-CSP等目标检测模型相比,模型在F1值上与YOLO v7相同,相比Scaled-YOLO v4、YOLOR-CSP分别提升0.7、0.2个百分点,在mAP0.5-0.95上分别提升0.6、1.2、0.2个百分点,而内存占用量仅为上述模型的14.2%、10.0%、10.0%。本文所提出的模型实现了小体量而高精度,便于在移动端进行部署,为后续机械化采摘和品质分级提供技术支持。

    Abstract:

    Sweet peppers are prone to malformed fruits during the growth and development process. Machine replace manual identification and removal of deformed sweet peppers, on the one hand, it can improve the quality and yield of sweet peppers;on the other hand, it can solve the current problems of high labor costs and low efficiency. In order to realize the identification of sweet pepper fruits by robots, an improved YOLO v7-tiny target detection model was proposed to distinguish between normal and abnormal growth of sweet pepper fruits. The parameterfree attention module (SimAM) was integrated into the backbone feature extraction network to enhance the feature extraction and feature integration capabilities of the model;the original loss function CIOU was replaced with Focal-EIOU loss, Focal-EIOU can speed up model convergence and reduce loss value;the SiLU activation function was used to replace the Leaky ReLU in the original network to enhance the nonlinear feature extraction ability of the model. The test results showed that the overall recognition precision, recall rate, mAP0.5 and mAP0.5-0.95 of the improved model were 99.1%, 97.8%, 98.9% and 94.5%, compared with that before improvement, it was increased by 5.4 percentage points, 4.7 percentage points, 2.4 percentage points, and 10.7 percentage points, respectively, the model weight size was 10.6MB, and the single image detection time was 4.2ms. Compared with YOLO v7, scaled-YOLO v4, YOLOR-CSP target detection models, the model had the same F1 score as YOLO v7. Compared with scaled-YOLO v4, YOLOR-CSP was increased by 0.7 and 0.2 percentage points, respectively, mAP0.5-0.95 was increased by 0.6 percentage points, 1.2 percentage points and 0.2 percentage points, respectively, and the weight size was only 14.2%, 10.0%, 10.0% of the above model. The model proposed achieved small size and high precision, and it was easy to deploy on the mobile terminal, providing technical support for subsequent mechanized picking and quality grading.

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

王昱,姚兴智,李斌,徐赛,易振峰,赵俊宏.基于改进YOLO v7-tiny的甜椒畸形果识别算法[J].农业机械学报,2023,54(11):236-246. WANG Yu, YAO Xingzhi, LI Bin, XU Sai, YI Zhenfeng, ZHAO Junhong. Malformed Sweet Pepper Fruit Identification Algorithm Based on Improved YOLO v7-tiny[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(11):236-246.

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