计算机工程

作者：王正家, 胡飞飞, 张成娟, 雷卓, 何涛

单位：1. 现代制造质量工程湖北省重点实验室, 湖北 武汉 4300682. 湖北工业大学机械工程学院, 湖北 武汉 430068

关键词：立体匹配,Transformer,自适应窗口,可分离自注意力机制,坐标注意力

基金：国家自然科学基金(51275158)

现有端到端的立体匹配算法为了减轻显存消耗和计算量而预设固定视差范围，在匹配精度和运行效率上难以平衡。提出一种基于轻量化Transformer的自适应窗口立体匹配算法。利用具有线性复杂度的坐标注意力层对低分辨率特征图进行位置编码，减轻计算量并增强相似特征的辨别力；设计轻量化Transformer特征描述模块，转换上下文相关的特征，并引入可分离多头自注意力层对Transformer进行轻量化改进，降低Transformer的延迟性；用可微匹配层对特征进行匹配，设计自适应窗口匹配细化模块进行亚像素级的匹配细化，在提高匹配精度的同时减少显存消耗；经视差回归后生成无视差范围的视差图。在KITTI2015、KITTI2012和SceneFlow数据集上的对比实验表明，该算法比基于标准Transformer的STTR在匹配效率上快了近4.7倍，具有更快的运行速度和更友好的存储性能；比基于3D卷积的PSMNet误匹配率降低了18%，运行时间快了5倍，实现了更好的速度和精度的平衡。

来源：2024年第2期

《计算机工程》期刊编辑部

查看计算机工程杂志2024年第2期