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

作     者：武星 齐泽宇 王龙军 杨俊杰 夏雪 WU Xing;QI Zeyu;WANG Longjun;YANG Junjie;XIA Xue

作者机构：南京航空航天大学机电学院南京210016 中国农业科学院农业信息研究所北京100081 

基　　金：国家自然科学基金项目(61973154) 国防基础科研计划项目(JCKY2018605C004) 中央高校基本科研业务费专项资金项目(NS2019033) 南京航空航天大学研究生创新基地(实验室)开放基金项目(kfjj20190516) 

出 版 物：《农业机械学报》 (Transactions of the Chinese Society for Agricultural Machinery)

年 卷 期：2020年第51卷第8期

页      码：17-25页

摘      要：为使苹果采摘机器人在复杂果树背景下能快速准确地检测出苹果,提出一种轻量化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%。

主 题 词：苹果 采摘机器人 目标检测 YOLOv3网络 深度可分离卷积 网络训练 

学科分类：1305[艺术学-设计学类] 13[艺术学] 081104[081104] 08[工学] 0804[工学-材料学] 081101[081101] 0811[工学-水利类] 

核心收录：

D　O　I：10.6041/j.issn.1000-1298.2020.08.002

馆 藏 号：203963537...