基于轻量化YOLOv3卷积神经网络的苹果检测方法

• 作者: 夏雪 杨俊杰 武星 王龙军 齐泽宇
• 来源期刊: 农业机械学报
• 学科: 工学
• 关键词: 苹果 采摘机器人 目标检测 YOLOv3网络 深度可分离卷积 网络训练
• 年，卷（期）: 2020,（8）
• 所属期刊栏目: 农业装备与机械化工程
• 页码范围: 17-25
• 页数: 9页
• 分类号: TP391.4
• 字数: 5848字
• 语种: 中文
• DOI: 10.6041/j.issn.1000-1298.2020.08.002

摘要

为使苹果采摘机器人在复杂果树背景下能快速准确地检测出苹果,提出一种轻量化YOLO(You only look once)卷积神经网络(Light-YOLOv3)模型与苹果检测方法.首先,对传统YOLOv3深度卷积神经网络架构进行改进,设计一种同构残差块串联的特征提取网络结构,简化目标检测的特征图尺度,采用深度可分离卷积替换普通卷积,提出一种融合均方误差损失和交叉熵损失的多目标损失函数;其次,开发爬虫程序,从互联网上获取训练数据并进行标注,采用数据增强技术对训练数据进行扩充,并对数据进行归一化,针对Light-YOLOv3网络训练,提出一种基于随机梯度下降(Stochastic gradient descent,SGD)和自适应矩估计(Adaptive moment estimation,Adam)的多阶段学习优化技术;最后,分别在计算机工作站和嵌入式开发板上进行了复杂果树背景下的苹果检测实验.结果 表明,基于轻量化YOLOv3网络的苹果检测方法在检测速度和准确率方面均有显著的提高,在工作站和嵌入式开发板上的检测速度分别为116.96、7.59 f/s,F1值为94.57％,平均精度均值(Mean average precision,mAP)为94.69％.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	武星	南京航空航天大学机电学院	44	435	13.0	19.0
2	王龙军	南京航空航天大学机电学院	7	36	2.0	6.0
3	夏雪	中国农业科学院农业信息研究所	11	68	5.0	8.0
4	齐泽宇	南京航空航天大学机电学院	1	0	0.0	0.0
5	杨俊杰	南京航空航天大学机电学院	1	0	0.0	0.0