PSMNet - 3.5 English

Vitis AI Library User Guide (UG1354)

Document ID
Release Date
3.5 English

PSMNet is a pyramid stereo matching network that can be used for depth estimation. It consists of two main modules: spatial pyramid pooling and 3D CNN. The spatial pyramid pooling module takes advantage of the capacity of global context information by aggregating context in different scales and locations to form a cost volume. The 3D CNN regularizes cost volume using stacked multiple hourglass networks with intermediate supervision.

The following image shows the result of PSMNet.

Figure 1. PSMNet Depth Estimation Example

The following table lists the PSMNet models supported by the Vitis AI Library.

Table 1. PSMNet Models
No Model Name Framework
1 PSMNet_0_int PyTorch
2 PSMNet_1_int PyTorch
3 PSMNet_2_int PyTorch