基于注意力改进CBAM的农作物病虫害细粒度识别研究
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金项目(61370102、61976052)和广东省基础与应用基础研究基金项目(2019B1515210009)


Fine-grained Identification Research of Crop Pests and Diseases Based on Improved CBAM via Attention
Author:
Affiliation:

Fund Project:

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

    预防和控制农作物病虫害是保证作物产量的重要措施。为了提高病虫害识别模型的准确率,对注意力CBAM模块进行改进,提出一种新的混合注意力模块I_CBAM。通过通道注意力与空间注意力的并行连接,解决了串行连接两种注意力产生干扰的问题。添加了I_CBAM模块的InRes-v2、MobileNet-v2、LeNet、AlexNet、改进AlexNet模型的Top-1(61类)准确率分别达到了86.98%、86.50%、80.97%、84.47%和84.96%,比原模型分别提高了0.51、0.62、1.74、0.53、0.55个百分点。研究表明,提出的并行混合注意力模块I_CBAM在病虫害细粒度分类上具有更优的识别效果,且在不同卷积神经网络模型之间拥有良好的泛化性。将I_CBAM中通道注意力压缩比调整为32,使添加了I_CBAM的MobileNet-v2迁移学习模型的内存缩小至28.3MB,预测一幅图像平均用时仅为7.19ms,大大提高了预测响应速度。将该模型部署到移动端小程序上,结果表明,添加了I_CBAM模块的模型具有良好的可视化应用效果。

    Abstract:

    Agricultural production is a significant part of Chinese economic development. The prevention and control of crop pests and diseases are critical measures to ensure crop yield. In order to improve the accuracy of the crop pests and diseases identification model, a new attention module I_CBAM improved from CBAM was proposed. By adopting a parallel connection structure of channel attention and spatial attention, the problem of interference caused by cascade of channel attention and spatial attention module was solved. By adding I_CBAM, the prediction accuracy of the model can be steadily improved. By adding I_CBAM to the five convolutional neural network models of InRes-v2, MobileNet-v2, LeNet, AlexNet, and improved AlexNet, the accuracy of Top-1 (61 types) reached 86.98%, 86.50%, 80.97%, 84.47% and 84.96%, respectively. Compared with the original model, it was improved by 0.51, 0.62, 1.74, 0.53 and 0.55 percentage points, respectively. The final results showed that the parallel mixed attention module I_CBAM proposed had better recognition effect on fine-grained classification of crop pests and diseases. And it also had good generalization in different other convolutional neural network models. Furthermore, by adjusting the channel attention ratio in I_CBAM to 32, the memory size of the MobileNet-v2 transfer learning model with I_CBAM was further reduced to 28.3MB. Meanwhile, the average time the model used to predict a picture was only 7.19ms, which made a good balance between the prediction cost and the prediction accuracy. Finally, the model was deployed on the mobile terminal mini application, which had a good visual application effect.

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

王美华,吴振鑫,周祖光.基于注意力改进CBAM的农作物病虫害细粒度识别研究[J].农业机械学报,2021,52(4):239-247. WANG Meihua, WU Zhenxin, ZHOU Zuguang. Fine-grained Identification Research of Crop Pests and Diseases Based on Improved CBAM via Attention[J]. Transactions of the Chinese Society for Agricultural Machinery,2021,52(4):239-247.

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