收录:
摘要:
本发明公开了一种基于感兴趣区域引导型注意力机制的图像分类方法。该图像方法包括:获取待分类图像和感兴趣区域掩膜图,计算空间重要性初始化图;将经过裁剪的待分类图像、经过裁剪的感兴趣区域掩膜图输入到特征提取模块中提取图像特征;将图像特征、空间重要性初始化图输入到空间注意力机制模块进行空间层面上的特征增强;将经过空间重要性加权后的特征、掩膜图输入到通道注意力模块进行通道层面的特征增强;将图像特征、掩膜图、空间重要性图进一步输入到特征提取模块、空间注意力模块和通道注意力模块中进行增强;将特征拉平,输入到全连接层中实现图像分类。本发明较现有方法特征表达能力更好,能够有效提升图像分类精准度。
关键词:
通讯作者信息:
电子邮件地址: