SSD Detection - 3.5 English

Vitis AI Library User Guide (UG1354)

Document ID
Release Date
3.5 English

The SSD Detection library is commonly used with the SSD neural network. SSD is a neural network that is used to detect objects. The input is a picture with some objects you want to detect. The output is a structure containing the coordinates and probability for each bounding box. The following image shows the result of SSD detection.

Figure 1. SSD Detection Example

The following table lists the SSD detection models supported by the Vitis AI Library.

Table 1. SSD Models
No Model Name Framework
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