基于改进SSD的柑橘实时分类检测

• 作者: 安小松 朱明 李善军 林家豪 胡定一 高淑敏
• 来源期刊: 农业工程学报
• 学科: 工学
• 关键词: 目标识别 模型 无损检测 柑橘 表面缺陷 深度学习 SSD ResNet18
• 年，卷（期）: 2019,（24）
• 所属期刊栏目: 农产品加工工程
• 页码范围: 307-313
• 页数: 7页
• 分类号: TP391.4
• 字数: 4573字
• 语种: 中文
• DOI: 10.11975/j.issn.1002-6819.2019.24.036

摘要

针对人工分拣柑橘过程中,检测表面缺陷费时费力的问题,该文提出了一种基于改进SSD深度学习模型的柑橘实时分类检测方法.在经改装的自制打蜡机试验台架下采集单幅图像含有多类多个柑橘的样本2500张,随机选取其中2000张为训练集,500张为测试集,在数据集中共有正常柑橘19507个,表皮病变柑橘9097个,机械损伤柑橘4327个.该方法通过单阶段检测模型SSD-ResNet18对图片进行计算和预测,并返回图中柑橘的位置与类别,以此实现柑橘的分类检测.以平均精度AP(average precision)的均值mAP(mean average precision)作为精度指标,平均检测时间作为速度指标,在使用不同特征图、不同分辨率和ResNet18、MobileNetV3、ESPNetV2、VoVNet39等4种不同特征提取网络时,进行模型分类检测效果对比试验研究.研究表明,该模型使用C4、C5特征图,768×768像素的分辨率较为合适,特征提取网络ResNet18在检测速度上存在明显优势,最终该模型的mAP达到87.89％,比原SSD的87.55％高出0.34个百分点,平均检测时间为20.27 ms,相较于原SSD的108.83 ms,检测耗时降低了436.90％.该模型可以同时对多类多个柑橘进行实时分类检测,可为自动化生产线上分拣表面缺陷柑橘的识别方面提供技术借鉴.