高级检索+

基于注意力机制的改进残差网络的柑橘分类研究

Research on Citrus Classification based on Improved Residual Network and Attention Mechanism

  • 摘要: 针对传统柑橘分级技术大多依赖人工获取特征信息,工作繁琐且效率低,难以实现在食品工业条件下进行有效识别的特点,提出一种基于注意力机制的改进残差网络的柑橘分类方法。该研究在残差网络(ResNet34)的基础上加入注意力机制,提高了有用信息的权重,同时降低了无关信息的权重,从而改善了分类模块的特征信息采集水平,进而提升了模型的分类能力。试验结果表明,基于注意力机制的残差网络对健康柑橘与缺陷柑橘的分类准确率达到99.02%,相较于原ResNet34模型分类准确率有了相对地提高和稳定。加入注意力机制的残差网络对于柑橘表面缺陷具有更好的特征提取能力,能够提取更多的柑橘缺陷特征信息。该研究有助于提高柑橘产业生产率,并为柑橘缺陷识别提供参考。

     

    Abstract: In view of the problem that traditional citrus classification methods mainly rely on manual extraction of features,which are complex and inefficient,and difficult to achieve efficient recognition in the food industry environment,an improved residual network citrus classification method based on attention mechanism is proposed.In this study, attention mechanism is added on the basis of residual network(ResNet34),which increases the weight of useful information, weakens the influence of irrelevant information,and improves the expression ability of network model and the classification ability of model.The experimental results show that the classification accuracy of the residual network based on attention mechanism for healthy and defective citrus reaches 99.02%,which is relatively improved compared with the original ResNet34 model.The residual network with attention mechanism has better feature extraction ability for citrus surface defects, and can extract more citrus defect feature information.This study helps to improve the productivity of citrus industry and provides a reference for citrus defect identification.

     

/

返回文章
返回