基于数据平衡和深度学习的开心果品质视觉检测方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(31872849)、山东省重点研发计划项目(2019GNC106037)、山东省高等学校青创计划团队项目(2020KJF004)和青岛市科技发展计划项目(19-6-1-66-nsh、19-6-1-72-nsh)


Pistachio Visual Detection Based on Data Balance and Deep Learning
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为探究数据集中分类数量的平衡性对开心果品质检测的影响,将开心果图像与深度学习网络相结合,提出一种数据自动平衡的检测方法。根据行业标准将开心果数据集分为开口、闭口和缺陷3类,在此基础上再分为未经数据平衡和经过数据平衡2个数据集,分别使用AlexNet、GoogLeNet、ResNet50、SqueezeNet、ShuffleNet和Xception 6种网络对2类数据集进行分类测试。结果表明,经过数据平衡的数据集网络准确率均得到了提高,6种网络平均测试准确率由96.75%提高到99.26%,SqueezeNet网络的测试集准确率提升最明显,由93.76%提高到99.02%,ResNet50网络的测试准确率最高,为99.96%。本文方法可用于开心果品质视觉检测。

    Abstract:

    “Happy or not” is an important content of pistachio quality detection. Combined with computer vision and deep learning network, a detection method of automatic data balance was proposed to explore the influence of data balance on pistachio quality detection. Firstly, a detection method of automatic data balance was proposed to explore the influence of data balance on pistachio quality detection. According to the industry standard, pistachio data sets were divided into three categories: open, closed and quality defect. Secondly, the data was formed into two data sets, one was the data set without data balance, the other was the data set after data balance. AlexNet, GoogLeNet, ResNet50, SqueezeNet, ShuffleNet and Xception were used to classify two kinds of datasets. The results showed that the accuracy of the network was improved after data balance, the average testing accuracy rate of six networks was increased from 96.75% to 99.26%. The accuracy rate of test set of SqueezeNet was improved the most obviously, which was from 93.76% to 99.02%, ResNet50 prediction accuracy rate was the highest, reached 99.96%. The rationality of network was verified by visualizing the location of high weight regions. The data balance method constructed had the same promotion value for other agricultural products quality detection, and also had a certain reference for other image classification projects.

    参考文献
    相似文献
    引证文献
引用本文

高霁月,倪建功,杨昊岩,韩仲志.基于数据平衡和深度学习的开心果品质视觉检测方法[J].农业机械学报,2021,52(7):367-372.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-09-13
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2021-07-10
  • 出版日期: