将传统深度学习的CBAM注意力机制算法直接用于害虫分类，得到的精度不理想，主要原因是害虫个体小、害虫与背景颜色差异小。为此，提出一种适用害虫分类的改进CBAM注意力机制算法。该算法通过改进通道注意力模块，提高害虫的关键信息特征在特征图中的比重，以解决害虫与背景颜色差异小的问题；改进空间注意力模块，以解决害虫个体小、感知难的问题。在消融实验中得到的结果表明，该算法能够有效解决害虫个体小、害虫与背景颜色差异小等问题，使分类准确率得到提升，达到75.9%。
科研之友
微信
新浪微博
Facebook
分享链接
