PSMNet は、深度推定に使用できるピラミッド ステレオ マッチング ネットワークです。空間ピラミッド プーリングと 3D CNN の 2 つの主要モジュールで構成されます。空間ピラミッド プーリング モジュールは、スケールと位置のさまざまに異なるコンテキストを集約してコスト量を形成することにより、グローバル コンテキスト情報の容量を利用します。3D CNN は Stacked Hourglass ネットワークと中間教師を使用してコスト量を正則化します。
次の画像は、PSMNet の結果を示しています。
図 1. PSMNet 深度推定の例

次の表に、Vitis AI ライブラリでサポートされる PSMNet モデルを示します。
番号 | モデル名 | フレームワーク |
---|---|---|
1 | PSMNet_0_int | PyTorch |
2 | PSMNet_1_int | PyTorch |
3 | PSMNet_2_int | PyTorch |