基于注意力机制的水果新鲜度检测可解释模型
作者:
作者单位:

(1.浙江工商大学 食药质量安全工程研究院 杭州 310018;2.浙江工商大学管理工程与电子商务学院 杭州 310018)

作者简介:

通讯作者:

中图分类号:

基金项目:

国家重点研发计划项目(2023YFD1000400);国家自然科学基金项目(62376249,91746202)


Fruit Freshness Detection Explainable Model Based on Attention Mechanism
Author:
Affiliation:

(1.Zhejiang Food and Drug Quality & Safety Engineering Research Institute,Zhejiang Gongshang University, Hangzhou 310018;2.School of Management and E-Business, Zhejiang Gongshang University, Hangzhou 310018)

Fund Project:

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

    近年来,基于机器视觉和深度学习的水果新鲜度检测成为主流方法之一。针对现有的深度学习技术,对卷积神经网络在水果的特征提取方面的应用进行探讨,在ResNet34主干网络中引入卷积注意力模块(CBAM),以实现水果新鲜度的检测,并采用类激活图(CAM)系列技术对于能够反映图片关键特征的像素进行热力图可视化。在水果公开数据集上,引入注意力机制前、后的ResNet34分类准确率分别为96.80%和99.71%。同时,CAM热力图反映注意力模型能够更加准确地捕获水果图像中变质腐烂的区域,表明提出的模型改善了深度学习特征提取的能力,不仅提高了模型的泛化能力,而且增强了模型的可解释性。

    Abstract:

    In recent years, fruit freshness detection based on machine vision and deep learning has become one of the mainstream methods. This study explores the application of deep learning technologies, particularly convolutional neural networks(CNNs), in feature extraction for fruit freshness identification. This paper introduces the CBAM (Convolutional block attention module) attention mechanism module into the ResNet34 (34-layer residual network) backbone network to achieve fruit freshness detection. Class activation mapping(CAM) techniques are employed to visualize the heatmaps of pixels that reflect the critical features in the images. On a public fruit dataset, the classification accuracy of the ResNet34 network before and after introducing the attention mechanism is 96.80% and 99.71%, respectively. The CAM heatmaps show that the attention model can more accurately capture the regions of interest in the fruit images, indicating that the proposed model improves the feature extraction capability of deep learning, not only enhancing the model's generalization ability but also increasing its interpretability.

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

张寅升,宋曾林,王海燕.基于注意力机制的水果新鲜度检测可解释模型[J].中国食品学报,2024,24(10):28-36

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-10-28
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-12-16
  • 出版日期:
文章二维码
版权所有 :《中国食品学报》杂志社     京ICP备09084417号-4
地址 :北京市海淀区阜成路北三街8号9层      邮政编码 :100048
电话 :010-65223596 65265375      电子邮箱 :chinaspxb@vip.163.com
技术支持:北京勤云科技发展有限公司

漂浮通知


×
《中国食品学报》杂志社招聘编辑