SA-Gate is a neural network that is used for indoor segmentation. The input is a pair images which are RGB image and HHA map generated with depth map. The output is a heat map where each pixels is predicted with a semantic category, like chair, bed, usual object in indoor.
The following image shows the result of SA-Gate segmentation.
Figure 1. SA-Gate segmentation Test Example
The following table lists the SA-Gate models supported by the Vitis AI library.
| No | Model Name | Framework |
|---|---|---|
| 1 | SA_gate_pt | PyTorch |