SSD 検出ライブラリは、一般的に SSD ニューラル ネットワークで使用されます。SSD は、オブジェクトを検出するために使用されるニューラル ネットワークです。入力は、検出対象となるオブジェクトを含む画像です。出力は、各境界ボックスの座標と確率を含む構造体です。次の画像は、SSD 検出の結果を示しています。
図 1. SSD 検出の例
次の表に、Vitis AI ライブラリでサポートされる 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 |