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.
| 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 |