«上一篇/Previous Article|本期目录/Table of Contents|下一篇/Next Article»

《江苏农业科学》[ISSN:1002-1302/CN:32-1214/S]

卷:

第51卷

期数:

2023年第15期

页码:

187-193

栏目:

农业工程与信息技术

出版日期:

2023-08-05

文章信息/Info

Title:

A real-time tomato recognition method based on improved YOLOv5s

作者:

杨国亮; 王吉祥; 聂子玲

江西理工大学电气工程与自动化学院，江西赣州 341000

Author(s):

Yang Guoliang; et al

关键词:

番茄检测; YOLOv5s; K-means++; GAM注意力模块; 加权双向特征金字塔

Keywords:

-

分类号:

TP391.41

DOI:

-

文献标志码:

A

摘要:

针对现有番茄检测精度低、没有品质检测和部署难度高等问题，提出基于YOLOv5s改进的番茄及品质实时检测方法，并与原始YOLOv5模型及其他经典模型进行对比研究。结果表明，针对番茄大小不同的问题，采用K-Means++算法重新计算先验锚框提高模型定位精度；在YOLOv5s主干网络末端添加GAM注意力模块，提升模型检测精度并改善鲁棒性；应用加权双向特征金字塔网络(BiFPN)修改原有结构，完成更深层次的加权特征融合；颈部添加转换器(transformer)，增强网络对多尺度目标的检测能力。改进后的YOLOv5s番茄识别算法检测速度达到72帧/s。在测试集中对番茄检测均值平均精度(mAP)达到93.9％，分别比SSD、Faster-RCNN、YOLOv4-Tiny、原始YOLOv5s模型提高17.2、13.1、5.5、3.3百分点。本研究提出的番茄实时检测方法，在保持检测速度的同时，可降低背景因素干扰，实现复杂场景下对番茄的精准识别，具有非常好的应用前景，为实现番茄自动采摘提供相应技术支持。

Abstract:

-

参考文献/References:

［1］李君明，项朝阳，王孝宣，等. “十三五”我国番茄产业现状及展望［J］. 中国蔬菜，2021(2)：13-20.
［2］王海楠，弋景刚，张秀花.番茄采摘机器人识别与定位技术研究进展［J］. 中国农机化学报，2020，41(5)：188-196.
［3］王文杰，贡亮，汪韬，等. 基于多源图像融合的自然环境下番茄果实识别［J］. 农业机械学报，2021，52(9)：156-164.
［4］阮承治，赵德安，陈旭，等. 双指型农业机器人抓取球形果蔬的控制器设计［J］. 中国农机化学报，2019，40(11)：169-175.
［5］陈科圻，朱志亮，邓小明，等. 多尺度目标检测的深度学习研究综述［J］. 软件学报，2021，32(4)：1201-1227.
［6］赵立新，邢润哲，白银光，等. 深度学习在目标检测的研究综述［J］. 科学技术与工程，2021，21(30)：12787-12795.
［7］包晓敏，王思琪.基于深度学习的目标检测算法综述［J］. 传感器与微系统，2022，41(4)：5-9.
［8］邵延华，张铎，楚红雨，等. 基于深度学习的YOLO目标检测综述［J］. 电子与信息学报，2022，44(10)：3697-3708.
［9］李萍，邵彧，齐国红，等. 基于跨深度学习模型的作物病害检测方法［J］. 江苏农业科学，2022，50(8)：193-199.
［10］Girshick R，Donahue J，Darrell T，et al. Rich feature hierarchies for accurate object detection and semantic segmentation［C］//2014 IEEE Conference on Computer Vision and Pattern Recognition.Columbus，2014：580-587.
［11］Girshick R.Fast R-CNN［C］//2015 IEEE International Conference on Computer Vision. Santiago，2016：1440-1448.
［12］Ren S Q，He K M，Girshick R，et al. Faster R-CNN：towards real-time object detection with region proposal networks［J］. IEEE Transactions on Pattern Analysis and Machine Intelligence，2017，39(6)：1137-1149.
［13］张文静，赵性祥，丁睿柔，等. 基于Faster R-CNN算法的番茄识别检测方法［J］. 山东农业大学学报(自然科学版)，2021，52(4)：624-630.
［14］龙洁花，赵春江，林森，等. 改进Mask R-CNN的温室环境下不同成熟度番茄果实分割方法［J］. 农业工程学报，2021，37(18)：100-108.
［15］Liu W，Anguelov D，Erhan D，et al. SSD：single shot multiBox detector［C］//European Conference on Computer Vision.Cham：Springer，2016：21-37.
［16］Redmon J，Divvala S，Girshick R，et al. You only look once：unified，real-time object detection［C］//2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas，NV，USA.IEEE，2016：779-788.
［17］文斌，曹仁轩，杨启良，等. 改进YOLOv3算法检测三七叶片病害［J］. 农业工程学报，2022，38(3)：164-172.
［18］张兆国，张振东，李加念，等. 采用改进YoloV4模型检测复杂环境下马铃薯［J］. 农业工程学报，2021，37(22)：170-178.
［19］黄彤镔，黄河清，李震，等. 基于YOLOv5改进模型的柑橘果实识别方法［J］. 华中农业大学学报，2022，41(4)：170-177.
［20］Dosovitskiy A，Beyer L，Kolesnikov A，et al. An image is worth 16x16 words：transformers for image recognition at scale［EB/OL］. 2020：arXiv：2010.11929.https：//arxiv.org/abs/2010.11929.
［21］Liu Y C，Shao Z R，Hoffmann N.Global attention mechanism：retain information to enhance channel-spatial interactions［EB/OL］. 2021：arXiv：2112.05561.https：//arxiv.org/abs/2112.05561.

相似文献/References:

[1]杜龙龙,朱景建,王荣扬,等.基于深度学习的水稻精量穴播排种系统设计与测试[J].江苏农业科学,2022,50(21):200.
　Du Longlong,et al.Design and test of rice precision hill planting and seed metering system based on deep learning[J].Jiangsu Agricultural Sciences,2022,50(15):200.
[2]周绍发,肖小玲,刘忠意,等.改进的基于YOLOv5s苹果树叶病害检测[J].江苏农业科学,2023,51(13):212.
　Zhou Shaofa,et al.Improved apple leaf disease detection based on YOLOv5s[J].Jiangsu Agricultural Sciences,2023,51(15):212.
[3]杨国亮,盛杨杨,洪鑫芳,等.基于改进YOLO v5s的果园番茄采摘检测方法[J].江苏农业科学,2024,52(24):187.
　Yang Guolang,et al.A detection method for orchard tomato picking based on improved YOLO v5s[J].Jiangsu Agricultural Sciences,2024,52(15):187.

备注/Memo

备注/Memo:

收稿日期：2022-11-28
基金项目：江西省教育厅科技计划(编号：GJJ190450、GJJ180484)。
作者简介：杨国亮(1973—)，男，江西宜春人，博士，教授，主要从事人工智能和模式识别研究。E-mail：ygliang30@126.com。
通信作者：王吉祥，硕士研究生，主要从事模式识别研究。E-mail：1661270181@qq.com。

常用功能

更新日期/Last Update: 2023-08-05