基于YOLO v5-Lite的自然环境木瓜成熟度检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(32071912)、广东省农业科技创新十大主攻方向“揭榜挂帅”项目(2022SDZG03)和广东省大学生科技创新培育专项资金项目(pdjh2023a0075)


Method of Maturity Detection for Papaya Fruits in Natural Environment Based on YOLO v5-Lite
Author:
Affiliation:

Fund Project:

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

    利用深度学习实现视觉检测技术对自然环境下树上木瓜成熟度的识别,从而监测木瓜生长期成熟度有重要意义。针对目前木瓜的成熟度主要以人工判断为主,缺乏对木瓜成熟度快速、准确的自动检测方法问题,本研究基于轻量化YOLO v5-Lite模型,对自然环境下木瓜成熟度检测方法进行研究,通过采集的1386幅木瓜图像,训练得到最优权值模型。实验结果表明,该模型对木瓜检测mAP为92.4%,与目前主流的轻量化目标检测算法YOLO v5s、YOLO v4-tiny以及两阶段检测算法Faster R-CNN相比,其mAP分别提高1.1、5.1、4.7个百分点;此外,在保证检测精度的前提下,检测时间为7ms,且模型内存占用量仅为11.3MB。同时,该模型对不同拍摄距离、不同遮挡情况、不同光照情况下的果实均能实现准确识别,能够快速有效地识别出复杂背景下木瓜果实的成熟度,具有较强的鲁棒性,可以为木瓜果园的产量估计和采摘机器的定位检测提供技术支持。

    Abstract:

    Using visual detection technologies based on deep learning to identify the maturity of papaya fruits on tree in natural environment and monitor the growing periods of papaya is of great significance to the intelligent management of papaya orchard. At present, there are relatively few studies on the identification of papaya maturity. The maturity of papaya is mainly judged manually, which have urgent needs to be replaced by some alternative fast and accurate automatic detection methods. Based on the lightweight YOLO v5-Lite model, a method of papaya maturity detection in natural environment was studied. The detection algorithm was improved based on the YOLO v5 network. In order to alleviate frequent slicing operations, a faster convolution operation was used to replace the Focus layer of the original network, which reduced the amount of computation and released the memory usage and accelerated the inference speed. To reduce the amount of calculation, the ShuffleNetv2 was used in the model to change the 1×1 group convolution in the middle to ordinary convolution by reducing the use of group convolution. At the same time, the ordinary convolution on the branch was changed to a depth-wise separable convolution, which greatly reduced the amount of calculation and improved the calculation efficiency. The number of C3 Layers especially the ones in deep neural blocks was reduced, so as to reduce the cache space occupation and speed up the operations. The channel number in FPN and PAN was set identical to speed the memory accessment. Totally 1386 papaya images were selected to create a dataset in PASCAL VOC format. Under the Ubuntu 16.04 environment, the training parameters of network were set as the epoch number of 300, the batchsize of 128, the total number of iterations of 300, and the initial learning rate of 0.001. During training, the loss value of the model tended to stabilize at the 200th iteration, indicating that the network was converged and the training performance was good. The evaluation indicators for papaya maturity identification of the experiments were the accuracy rate, recall rate, overall average accuracy, detection speed and model size. The experimental results showed that the mAP of the papaya maturity detection model was 92.4%, which outperformed the mainstream lightweight object detection algorithms namely the YOLO v5s and the YOLO v4-tiny and the classic two-stage algorithm Faster R-CNN by 1.1 percentage points, 5.1 percentage points and 4.7 percentage points, on mAP, respectively. In addition, under the condition of relatively accurate detection, the detection time was up to 7ms, and the model size was only 11.3MB. At the same time, the model can accurately identify the fruits under different shooting distances, occlusion conditions and lighting conditions, showing the performance of fast and effective identification and good robustness under complex backgrounds. The proposed method provided technical support for yield estimation of papaya orchards and the positioning detection of picking robots, which can also provide reference for researches on the maturity detection of other fruits.

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

熊俊涛,韩咏林,王潇,李泽星,陈浩然,黄启寅.基于YOLO v5-Lite的自然环境木瓜成熟度检测方法[J].农业机械学报,2023,54(6):243-252. XIONG Juntao, HAN Yonglin, WANG Xiao, LI Zexing, CHEN Haoran, HUANG Qiyin. Method of Maturity Detection for Papaya Fruits in Natural Environment Based on YOLO v5-Lite[J]. Transactions of the Chinese Society for Agricultural Machinery,2023,54(6):243-252.

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