典型文献
基于改进YOLO及NMS的水果目标检测
文献摘要:
为使水果采摘机器人在复杂情况下如树叶遮挡、果实目标尺度变化大等情况能准确地检测出水果,提出一种YOLO(you only look once)改进模型与NMS(non-maximum suppression)改进算法的目标检测方法.首先,对传统YOLO深度卷积神经网络架构进行改进,设计一种更细化的SPP5(spatial pyramid pooling)特征融合网络模块,强化特征图多重感受野信息的融合,并基于此模块提出一种YOLOv4-SPP2-5模型,在标准YOLOv4网络中跨层添加并改进SPP层,重新分布池化核大小,增强感受野范围,从而降低目标误检率;其次,提出一种Greedy-Confluence的NMS改进算法,通过对高度接近的检测框直接抑制和对重叠检测框综合考虑距离交并比DIOU(distance-intersection over union)和加权接近度WP(weighted proximity)的方法,均衡NMS的计算消耗并减少检测框的错误抑制,从而提高遮挡、重叠物体的检测精度;最后,分别对改进方法进行性能测试,验证方法的可行性,随后制作水果检测数据集并进行格式转换和标签标注,然后采用数据增强技术对训练数据进行扩充,并使用K-means++聚类方法获取先验锚定框,在计算机上进行了水果检测实验.结果表明,基于改进YOLO网络及改进NMS的水果检测方法在准确率方面有显著的提高,平均精度均值(mean average precision,MAP)在YOLOv4上达到了 96.65%,较原网络提升1.70%,并且实时性也得到了保证,在测试设备上达到了 39.26帧/s.
文献关键词:
水果目标检测;YOLO网络;SPP模块;NMS;信息熵
中图分类号:
作者姓名:
徐印赟;江明;李云飞;吴云飞;卢桂馥
作者机构:
安徽工程大学高端装备先进感知与智能控制教育部重点实验室 芜湖 241000;安徽工程大学电气工程学院 芜湖 241000;安徽工程大学计算机与信息学院 芜湖 241000
文献出处:
引用格式:
[1]徐印赟;江明;李云飞;吴云飞;卢桂馥-.基于改进YOLO及NMS的水果目标检测)[J].电子测量与仪器学报,2022(04):114-123
A类:
水果目标检测,SPP5
B类:
NMS,水果采摘机器人,复杂情况,树叶,遮挡,标尺,尺度变化,you,only,look,once,改进模型,maximum,suppression,改进算法,目标检测方法,深度卷积神经网络,神经网络架构,spatial,pyramid,pooling,特征融合网络,特征图,YOLOv4,SPP2,中跨,跨层,新分布,池化,增强感受野,误检率,Greedy,Confluence,距离交并比,DIOU,distance,intersection,over,union,WP,weighted,proximity,计算消耗,检测精度,改进方法,验证方法,水果检测,检测数据集,格式转换,标签标注,数据增强技术,训练数据,means++,聚类方法,先验,锚定框,平均精度均值,average,precision,MAP,上达,测试设备,信息熵
AB值:
0.474775
相似文献
机标中图分类号,由域田数据科技根据网络公开资料自动分析生成,仅供学习研究参考。