摘要
本发明公开了一种基于共享特征和注意力上采样的双目视差匹配方法及系统,该方法步骤包括:将左、右图像预处理后提取1/2尺度、1/4尺度、1/8尺度、1/16尺度特征图;左、右图像1/16尺度特征图构建匹配代价矩阵,生成1/16尺度的初始视差图;使用1/16尺度初始视差图和1/16尺度图像特征图估计出1/16尺度的视差残差图,利用注意力机制实现上采样,生成1/8尺度的视差图;分别使用1/8尺度、1/4尺度、1/2尺度视差图和对应的图像特征图,分别生成1/4尺度、1/2尺度和原尺度视差图;进行模型训练并保存最优模型参数;载入预训练参数,输入图像帧,获取不同尺度视差图。本发明采用共享特征设计和注意力机制上采样,有效提升了视差匹配的精度和速度,能够实时地生成高精度视差图。
- 单位