PSMNet 是一种金字塔立体匹配网络,可用于深度估计。它包含 2 个主要模块:空间金字塔池化模块和三维卷积神经网络模块。空间金字塔池化模块通过汇总不同比例和位置中的上下文构成成本量来充分利用丰富的全局上下文信息。三维卷积神经网络使用堆叠多个沙漏模型搭配中间监督来将成本量加以正则化。
下图显示了 PSMNet 的结果。
图 1. PSMNet 深度估算示例

下表列出了 Vitis AI Library 支持的 PSMNet 模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | PSMNet_0_int | PyTorch |
2 | PSMNet_1_int | PyTorch |
3 | PSMNet_2_int | PyTorch |