基于YOLO-FFD的水果品种和新鲜度识别方法
CSTR:
作者:
作者单位:

(武汉轻工大学电气与电子工程学院,湖北 武汉 430023)

作者简介:

鄢紫,女,武汉轻工大学在读硕士研究生。

通讯作者:

陈良艳(1982—),女,武汉轻工大学副教授,博士。E-mail:chenliangyan@whpu.edu.cn

中图分类号:

基金项目:

湖北省高校优秀中青年科技创新团队项目(编号:T2021009)


Fruit variety and freshness recognition method based on YOLO-FFD
Author:
Affiliation:

(School of Electronical and Electronic Engineering, Wuhan Polytechnic University, Wuhan, Hubei 430023, China)

Fund Project:

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

    目的:改善现有水果识别与分级方法依赖于人工操作和复杂设备的情况。方法:提出了一种轻量化模型YOLO-FFD(YOLO with fruit and freshen detection),该模型以YOLOv5框架为基础,基于深度可分离卷积和GELU激活函数设计轻量化模块LightweightC3作为主干特征提取网络的基本单元,减少模型参数量和计算量,加快模型的收敛速度;使用大内核深度可分离卷积模块EnhancedC3改进原模型的颈部,抑制信息丢失并增强模型的特征融合能力,提高模型的检测精度;采用GSConv代替特征融合网络中的普通卷积,使模型进一步轻量化。结果:提出模型的平均精度均值达到了96.12%,在RTX 3090上速度为172帧/s,在嵌入式设备Jetson TX2上速度为20帧/s。相比于原始YOLOv5模型,平均精度均值提高了2.21%,计算量减少了26%,在RTX 3090和Jetson TX2上的速度分别提高了2倍和1倍。结论:YOLO-FFD模型能够满足识别水果品种和新鲜度的需求,且在复杂场景下错检、漏检情况均有改善。

    Abstract:

    Objective: In order to improve the situation that existing fruit recognition and classification methods rely on manual operation and complex equipment. Methods: A lightweight model YOLO-FFD (YOLO with fruit and freshen detection) was proposed, which based on the YOLOv5 framework. Firstly, LightweightC3 was designed as the basic unit of the backbone feature extraction network based on the depth separable convolution and GELU activation function, which reduced the number of model parameters and computation, and speeds up the convergence of the model. Secondly, EnhancedC3, a large kernel depth separable convolution module, was used to improve the neck of the original model, suppressed information loss and enhance the feature fusion ability of the model, so as to improve the detection accuracy of the model. Finally, GSConv was used to replace the common convolution in the feature fusion network to further lighten the model. Results: The experimental results showed that the average accuracy of the proposed model reached 96.12%, the FPS on RTX 3090 was 172, and the speed on the embedded Jetson TX2 was 20 frames per second. Compared with the original YOLOv5 model, the mAP was improved by 2.21%, the calculation amount was reduced by 26%, and the speed was increased by two times. Conclusion: YOLO-FFD can meet the requirement of identifying fruit varieties and freshness, and improve the falsely detection and missing detection in complex scenes.

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

鄢 紫,陈良艳,刘卫华,等.基于YOLO-FFD的水果品种和新鲜度识别方法[J].食品与机械,2024,40(1):115-121.
YAN Zi, CHEN Liangyan, LIU Weihua, et al. Fruit variety and freshness recognition method based on YOLO-FFD[J]. Food & Machinery,2024,40(1):115-121.

复制
相关视频

分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-05-15
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-01-30
  • 出版日期:
文章二维码
×
《食品与机械》
友情提示
友情提示 一、 近日有不少作者反应我刊官网无法打开,是因为我刊网站正在升级,旧网站仍在百度搜索排名前列。请认准《食品与机械》唯一官方网址:http://www.ifoodmm.com/spyjx/home 唯一官方邮箱:foodmm@ifoodmm.com; 联系电话:0731-85258200,希望广大读者和作者仔细甄别。