公开(公告)号：CN118298461A

公开(公告)日：2024-07-05

申请号：CN202410408360.9

申请日：2024-04-07

申请人： 安徽工程大学 ,  安徽工程大学产业创新技术研究有限公司

发明人： 徐晓峰 ,  丁伯宇 ,  包象琳 ,  赵薇 ,  杜文龙 ,  戴家树 ,  严楠

IPC分类号： G06V40/10 ,  G06V10/44 ,  G06V10/764 ,  G06V10/82 ,  G06N3/0464

摘要： 本发明涉及图像分类技术领域，具体涉及一种注意力二叉导航树的细粒度图像分类方法，包括建立注意力二叉导航树模型，所述注意力二叉导航树模型包括主干网络模块、注意力模块、分支路由模块、局部特征注意力学习模块、导航模块和预测模块，将获得的第一个至第四个叶子节点的概率分布乘以对应路径分支路由模块输出的权重并求和，得到最终预测结果的概率分布，将图片分类到最终预测结果的概率分布中最大概率值对应的图像类别中。本方法减少了图片背景部分的影响，在整体上实现了以更合理的方式分配每个阶段的权重，帮助模型提高分类的能力。