一种基于空间注意力的群等变胶囊网络的设计方法

本发明公开了一种基于空间注意力的群等变胶囊网络的设计方法，包括：S1、构建具有等变性的主干网络，从输入图片中提取群等变的特征图；S2、使用全局平均池化和最大池化相结合的方式处理群等变的特征图，得到具有不变性的胶囊描述子；S3、使用群等变的空间注意力机制加强特征图中空间信息，得到群等变的胶囊姿态；S4、使用一种平均最大池化对胶囊姿态进行处理，方便使用语义分割任务验证胶囊姿态的等变性；S5、将胶囊描述子和姿态中的信息进行交互，提升网络在分类和分割任务中的性能。根据本发明，能提供可以数学证明的等变性，并且模型参数量较小，适用于输入图像会发生仿射变化的情况，可直接应用于实际中的分类和语义分割任务。