一种轻量型热图立体匹配方法及装置

本发明公开了一种轻量型热图立体匹配方法及装置，用于解决现有的轻量型热图立体匹配方法在捕捉细微深度差异时容易出现信息丢失，导致最终的视差结果的精度欠佳的技术问题。方法包括获取左目、右目热图像，输入预置热图像立体匹配模型（含多尺度融合特征提取、边缘特征提取、增强型小波注意力模块及多个代价聚合残差块、卷积模块）。多尺度模块提取特征，生成左目与右目融合特征图，构建初始代价体；边缘模块输出左目边缘概率特征图，结合初始代价体得中间代价体。小波注意力模块以左目融合特征图输出注意力权重；代价聚合残差块依权重与中间代价体相乘，经视差回归生成初始视差图；卷积模块结合左目边缘概率特征图细化，最终生成目标视差图。