嵌入式设备的轻量化百香果检测模型
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

海峡博士后交流资助计划项目


Lightweight Passion Fruit Detection Model Based on Embedded Device
Author:
Affiliation:

Fund Project:

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

    为在有限的嵌入式设备资源下达到实时检测要求,提出一种基于改进YOLO v5的百香果轻量化检测模型(MbECA-v5)。首先,使用MobileNetV3替换主干特征提取网络,利用深度可分离卷积代替传统卷积减少模型的参数量。其次,嵌入有效通道注意力网络(ECANet)关注百香果整体,引入逐点卷积连接特征提取网络和特征融合网络,提高网络对百香果图像的特征提取能力和拟合能力。最后,运用跨域与域内多轮训练相结合的迁移学习策略提高网络检测精度。试验结果表明,改进后模型的精确率和召回率为95.3%和88.1%;平均精度均值为88.3%,较改进前提高0.2个百分点。模型计算量为6.6 GFLOPs,体积仅为6.41MB,约为改进前模型的1/2,在嵌入式设备实时检测速度为10.92f/s,约为SSD、Faster RCNN、YOLO v5s模型的14倍、39倍、1.7倍。因此,基于改进YOLO v5的轻量化模型提高了检测精度和大幅降低了计算量和模型体积,在嵌入式设备上能够高效实时地对复杂果园环境中的百香果进行检测。

    Abstract:

    In order to meet the real-time detection requirements under the limited resources of embedded devices, a passion fruit detection model based on improved YOLO v5 lightweight network (MbECA-v5) was proposed. Firstly, MobileNetV3 was used to replace the feature extraction network, the depth separable convolution was used to replace the traditional convolution to reduce the number of model parameters. Secondly, the effective channel attention network (ECANet) was embedded to focus on the whole passion fruit. Point-by-point convolution connection feature extraction network and feature fusion network were introduced to improve the feature extraction ability and fitting ability of the network for passion fruit images. Finally, the transfer learning strategy combined with cross-domain and within-domain multi-training was used to improve the network detection accuracy. Experimental results showed that the accuracy and recall of the improved model were 95.3% and 88.1%, respectively. The mAP value of 88.3%,compared with the model before the improvement, it was increased by 0.2 percentage points. And the number of calculations was 6.6 GFLOPs. The model volume was only 6.41MB, which was about half of the improved model. The real-time detection speed in embedded device was 10.92f/s, the detection speed in embedded device was about 14 times,39 times and 1.7 times of SSD, Faster RCNN and YOLO v5s. Therefore, the lightweight model based on improved YOLO v5 greatly reduced the amount of calculation and model volume, and it can detect passion fruit in complex orchard environment efficiently on embedded devices, which was of great significance to improve the intelligent level of orchard production.

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

罗志聪,李鹏博,宋飞宇,孙奇燕,丁昊凡.嵌入式设备的轻量化百香果检测模型[J].农业机械学报,2022,53(11):262-269,322.

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