多階段注意力膠囊網絡的圖像分類

摘要： 針對傳統的膠囊網絡(Capsule network, CapsNet)特征提取不充分的問題,提出一種圖像分類的多階段注意力膠囊網絡模型.首先,在卷積層對低層特征和高層特征分別采用注意力(Spatial attention, SA)和通道注意力(Channel attention, CA)來提取有效特征;然后,提出基于向量的注意力(Vector attention, VA)機制作...