“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_pt | PyTorch |