基于多尺度双路注意力胶囊网络在水稻害虫识别中的应用

[目的]研究基于多尺度双路注意力胶囊网络(Multi-scale Dual-path Attention Capsule Network,MDACapsNet)的水稻害虫识别方法,为田间害虫自动识别提供参考依据.[方法]针对现有方法对位置、姿势多变的水稻害虫识别准确率低等问题,在胶囊网络(Capsule Network,CapsNet)的基础上,提出一种基于MDACapsNet的水稻害虫识别方法.MDACapsNet由编码模块、重构模块与分类模块组成.在MDACapsNet编码模块中,多尺度双路注意力模块与局部共享动态路由算法用于提高网络的特征提取能力并降低网络计算量;在MDACapsNet的损失函数中增加重构图像损失项,加速网络训练.[结果]在复杂背景下的水稻害虫图像数据集进行实验,识别准确率为95.31％.与VGG16、CapsNet、ResNet相比较,MDACapsNet的识别准确率分别提高5.47％、19.83％、2.83％.[结论]MDACapsNet能较好地识别大小不同、背景复杂、姿势多样的水稻害虫,可应用于田间水稻害虫自动检测系统中.