車両分類ライブラリは、車両画像を自動車メーカーや車両タイプで分類するために使用されます。これらのニューラル ネットワークは、CompCars でトレーニング済みであり、163 個または 12 個のクラスから物体を識別できます。Vitis AI ライブラリは、vehicle_make_resnet18_pt および vehicle_type_resnet18_pt を含むネットワークをAMD ライブラリに統合しています。入力はオブジェクトを含む画像で、出力は上位 K 番目までの最も可能性の高いカテゴリとなります。
図 1. 車両分類の例
番号 | モデル名 | フレームワーク |
---|---|---|
1 | vehicle_type_resnet18_pt | PyTorch |
2 | vehicle_make_resnet18_pt | PyTorch |