首页 | 本学科首页   官方微博 | 高级检索  
     

基于轻量化YOLOv3卷积神经网络的苹果检测方法
引用本文:武星,齐泽宇,王龙军,杨俊杰,夏雪. 基于轻量化YOLOv3卷积神经网络的苹果检测方法[J]. 农业机械学报, 2020, 51(8): 17-25
作者姓名:武星  齐泽宇  王龙军  杨俊杰  夏雪
作者单位:南京航空航天大学机电学院,南京210016;中国农业科学院农业信息研究所,北京100081
基金项目:国家自然科学基金项目(61973154)、国防基础科研计划项目(JCKY2018605C004)、中央高校基本科研业务费专项资金项目(NS2019033)和南京航空航天大学研究生创新基地(实验室)开放基金项目(kfjj20190516)
摘    要:为使苹果采摘机器人在复杂果树背景下能快速准确地检测出苹果,提出一种轻量化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网络  深度可分离卷积  网络训练
收稿时间:2019-12-03

Apple Detection Method Based on Light-YOLOv3 Convolutional Neural Network
WU Xing,QI Zeyu,WANG Longjun,YANG Junjie,XIA Xue. Apple Detection Method Based on Light-YOLOv3 Convolutional Neural Network[J]. Transactions of the Chinese Society for Agricultural Machinery, 2020, 51(8): 17-25
Authors:WU Xing  QI Zeyu  WANG Longjun  YANG Junjie  XIA Xue
Affiliation:Nanjing University of Aeronautics and Astronautics; Agricultural Information Institute, Chinese Academy of Agricultural Sciences
Abstract:
Keywords:apple   picking robot   object detection   YOLOv3 network   depth wise separable convolution   network training
本文献已被 CNKI 万方数据 等数据库收录!
点击此处可从《农业机械学报》浏览原始摘要信息
点击此处可从《农业机械学报》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号