公开(公告)号：CN118196533A

公开(公告)日：2024-06-14

申请号：CN202410406742.8

申请日：2024-04-07

Applicant: 北京林业大学

Inventor： 张军国 ,  王子腾 ,  丁保琛 ,  赵一冲 ,  李懿轩

IPC: G06V10/764 ,  G06V10/774 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了基于注意力机制和ResNet50网络的野生动物图像分类方法，涉及野生动物分类技术领域。本发明与之前的野生动物分类方法相比，解决了传统深度网络在测试数据集是均匀分布的情况下，在长尾分布数据集上的表现效果不佳的问题，将ResNet50与Class Balanced Loss相结合，以平衡各个类别在训练中的权重，使得模型更加关注少数类别的样本，从而提高在长尾数据上的性能；并引入SGE注意力机制在模型的特定层引入空间注意力增强，通过使得模型在学习过程中更加关注图像中的重要结构信息，提高模型对于关键区域的感知能力。通过上述综合分类方法，在保持传统网络特性的同时，有效提升模型性能。