摘要
本发明公开了一种细粒度小样本图像的分类方法及装置,包括第一阶段网络即目标定位网络,包括相互连接的自注意力互补定位模块及VGG-16网络,用于训练一个弱监督目标定位网络,得到查询图像和支持图像的类激活图;第二阶段网络包括特征提取模块、特征描述子筛选模块及语义对齐距离模块;所述特征提取模块,用于分别对查询图像和支持图像进行特征提取,得到其对应的特征图;特征描述子筛选模块,根据类激活图和特征提取模块提取的特征图,生成查询图像和支持图像的特征描述子表示;语义对齐距离模块获得查询图像和支持图像之间的语义对齐距离;本发明能实现高精度的细粒度小样本图像分类。
- 单位