一种基于最优传输的自适应特征融合方法及装置

本发明属于深度学习与计算机视觉领域，公开了一种基于最优传输的自适应特征融合方法及装置，该方法包括：获取待融合的第一图像特征图和第二图像特征图；将两个图像特征图分别在空间维度上展平，得到第一特征向量集合和第二特征向量集合；基于两个特征向量集合中特征向量之间的配对距离，构建传输成本矩阵；对传输成本矩阵进行迭代优化，直至得到最优传输方案矩阵；通过传输成本矩阵和最优传输方案矩阵得到瓦瑟斯坦距离，并将瓦瑟斯坦距离映射为自适应融合门控权重；通过自适应融合门控权重融合图像特征图。本发明基于最优传输理论的自适应特征融合方法，以实现有原则的、分布感知的且更为智能的特征融合机制，提升了融合后的图像特征图的准确度。