高级检索+

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

Pistachio Visual Detection Based on Data Balance and Deep Learning

  • 摘要: 为探究数据集中分类数量的平衡性对开心果品质检测的影响,将开心果图像与深度学习网络相结合,提出一种数据自动平衡的检测方法。根据行业标准将开心果数据集分为开口、闭口和缺陷3类,在此基础上再分为未经数据平衡和经过数据平衡2个数据集,分别使用AlexNet、GoogLeNet、ResNet50、SqueezeNet、ShuffleNet和Xception6种网络对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.

     

/

返回文章
返回