基于深度学习水果检测的研究与改进

• 作者: 段先华 黄欣辰 黄豪杰
• 来源期刊: 计算机工程与应用
• 学科: 工学
• 关键词: 深度学习 目标检测 SSD模型 ResNet-101模型 特征金字塔网络(FPN)
• 年，卷（期）: 2020,（3）
• 所属期刊栏目: 模式识别与人工智能
• 页码范围: 127-133
• 页数: 7页
• 分类号: TP391
• 字数: 5098字
• 语种: 中文
• DOI: 10.3778/j.issn.1002-8331.1906-0384

摘要

为实现自然环境下水果自动化采摘存在受环境和障碍物等因素造成的问题,导致目标水果检测准确率不高,泛化性不强等实际问题,以苹果、橘子、香蕉三种水果作为研究对象,提出一种基于深度学习的SSD(Single Shot Detector)改进模型.经典SSD采用多尺度特征融合的方式,从网络不同层抽取不同尺度的特征做预测,但是没有用到足够低层的特征,使得小物体的检测效果较差.通过将经典SSD训练使用的VGG16输入模型替换为ResNet-101,利用特征金字塔网络(FPN)结构将高层特征通过上采样和低层特征做融合.实验表明,改进的SSD300和SSD512水果检测模型的平均检测精度为83.05％和84.24％,经数据增强后精度也有所提升,适合于自然环境下水果的精确检测.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	段先华	江苏科技大学计算机学院	42	192	9.0	11.0
2	黄欣辰	江苏科技大学计算机学院	2	3	1.0	1.0
3	黄豪杰	江苏科技大学计算机学院	1	3	1.0	1.0