基于多尺度双路注意力胶囊网络在水稻害虫识别中的应用
2022-05-20分类号:TP391.41;S435.112
【部门】西京学院信息工程学院 宝鸡市农业科学研究院
【摘要】【目的】田间水稻害虫识别是精确防治水稻害虫的重要前提。由于水稻害虫种类多,同种害虫形状大小、颜色、分布变化多样,且田间环境背景复杂,使得传统方法与基于卷积神经网络的方法识别准确率较低。【方法】针对现有方法对位置、姿势多变的水稻害虫识别准确率低等问题,在胶囊网络(Capsule Network, CapsNet)的基础上,提出一种基于多尺度双路注意力胶囊网络(Multi-scale Dual-path Attention Capsule Network , MDACapsNet)的水稻害虫识别方法。MDACapsNet由编码模块、重构模块与分类模块组成。在MDACapsNet编码模块中,多尺度双路注意力模块与局部共享动态路由算法用于提高网络的特征提取能力并降低网络计算量;在MDACapsNet的损失函数中增加重构图像损失项,加速网络训练。【结果】在复杂背景下的水稻害虫图像数据集进行实验,识别准确率为95.31%。与VGG16、CapsNet、ResNet相比较,MDACapsNet的识别准确率分别提高5.47%、19.83%、2.83%。【结论】MDACapsNet能较好地识别大小不同、背景复杂、姿势多样的水稻害虫,可应用于田间水稻害虫自动检测系统中。
【关键词】水稻害虫图像 胶囊网络 自注意力机制 多尺度双路注意力胶囊网络
【基金】国家自然科学基金项目(62072378)
【所属期刊栏目】西南农业学报
文献传递