语义分割会向输入图像中的每个像素分配一个语义类别,即,它把像素分类为对象(例如,一辆汽车、一条道路、一棵树、一匹马等)的一部分。Libsegmentation 是可在 ADAS 应用中使用的分割库。它可提供简单的接口,供开发者用于在 AMD 目标上部署分割任务。
以下是语义分割的示例,其中“蓝灰色”表示天空,“绿色”表示树木,“红色”表示人,“深蓝色”表示车辆,“绛紫色”表示道路,“灰色”表示结构。
图 1. 语义分割示例
下表列出了 Vitis AI Library 支持的语义分割模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | fpn | Caffe |
2 | FPN-resnet18_Endov | |
3 | semantic_seg_citys_tf2 | TensorFlow |
4 | mobilenet_v2_cityscapes_tf | |
5 |
SemanticFPN_cityscapes_pt |
PyTorch |
6 | ENet_cityscapes_pt | |
7 | unet_chaos-CT_pt | |
8 | SemanticFPN_Mobilenetv2_pt | |
9 | HRNet_pt |