“SSD Detection”（SSD 检测）库一般搭配 SSD 神经网络一起使用。SSD 是用于对象检测的神经网络。输入是含待测对象的图像。输出是包含每个边界框的坐标和概率的结构。下图显示了 SSD 检测的结果。
图 1. SSD 检测示例
下表列出了 Vitis AI Library 支持的 SSD 检测模型。
| 编号 | 模型名称 | 框架 |
|---|---|---|
| 1 | ssd_mobilenet_v1_coco_tf | TensorFlow |
| 2 | ssd_mobilenet_v2_coco_tf | |
| 3 | ssd_resnet_50_fpn_coco_tf | |
| 4 | mlperf_ssd_resnet34_tf | |
| 5 | ssdlite_mobilenet_v2_coco_tf | |
| 6 | ssd_inception_v2_coco_tf | |
| 7 | ssd_pedestrian_pruned_0_97 | Caffe |
| 8 | ssd_traffic_pruned_0_9 | |
| 9 | ssd_adas_pruned_0_95 | |
| 10 | ssd_mobilenet_v2 |