科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于注意力机制的视差优化方法,使用视差预估主干网络对双目视觉图像进行初始视差值计算,得到预测视差值,所述双目视觉图像包括左图像及右图像；设置多层次信息提取子网络,将注意力机制嵌入到子网络中,将左图像输入后,得到两幅不同维度的特征图；将子网络提取的特征图与视差预估主干网络中不同位置的特征图在对应维度上合并,进行细化视差值计算；采用监督回归损失函数端对端训练网络,所述网络包括视差预估主干网络和多层次信息提取子网络；将待预测的双目视觉图像输入完成训练的网络中,得到对应的视差预测值。本方法提升视差预估结果的精确度和鲁棒性。
- 单位
