SA-Gate 分割 - 3.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2023-06-29
Version
3.5 简体中文

“SA-Gate”(分合门)是用于室内分割的神经网络。输入是一对 RGB 图像和搭配深度映射生成的 HHA 映射。输出是热图,其中每个像素均以一个语义类别(例如,椅子、床和其他常见室内物体)来预测。

下图显示了 SA-Gate 分割的结果。

图 1. SA-Gate 分割测试示例 1

图 2. SA-Gate 分割测试示例 2

图 3. SA-Gate 分割测试示例 3

下表列出了 Vitis AI Library 支持的 SA-Gate 模型。

表 1. SA-Gate 分割模型
编号 模型名称 框架
1 SA_gate_pt PyTorch