分類 - 2.5 日本語

Vitis AI ライブラリ ユーザー ガイド (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 日本語

分類ライブラリは、イメージを分類するために使用されます。これらのニューラル ネットワークは ImageNet データセットの 1000 クラス ILSVRC サブセットでトレーニングされており、画像内の物体が各クラスのメンバーである確率をクラスごとに予測します。Vitis AI ライブラリは、ResNet18、ResNet50、Inception_v1、Inception_v2、Inception_v3、Inception_v4、VGG、mobilenet_v1、mobilenet_v2、Squeezenet を含む (ただし、これらに限定されない) 分類ネットワークをサポートしており、これらはザイリンクス ライブラリに統合されています。入力はオブジェクトを含む画像で、出力は上位 K 番目までの最も可能性の高いカテゴリとなります。

図 1. 分類の例

次の表に、Vitis AI ライブラリでサポートされる分類モデルを示します。

表 1. 分類モデル
番号 モデル名 フレームワーク
1 inception_resnet_v2_tf TensorFlow
2 inception_v1_tf
3 inception_v3_tf
4 inception_v4_2016_09_09_tf
5 mobilenet_v1_0_25_128_tf
6 mobilenet_v1_0_5_160_tf
7 mobilenet_v1_1_0_224_tf
8 mobilenet_v2_1_0_224_tf
9 mobilenet_v2_1_4_224_tf
10 resnet_v1_101_tf
11 resnet_v1_152_tf
12 resnet_v1_50_tf
13 vgg_16_tf
14 vgg_19_tf
15 mobilenet_edge_1_0_tf
16 mobilenet_edge_0_75_tf
17 inception_v2_tf
18 MLPerf_resnet50_v1.5_tf
19 resnet50_tf2
20 mobilenet_1_0_224_tf2
21 inception_v3_tf2
22 resnet_v2_50_tf
23 resnet_v2_101_tf
24 resnet_v2_152_tf
25 efficientnet-b0_tf2
26 efficientNet-edgetpu-S_tf
27 efficientNet-edgetpu-M_tf
28 efficientNet-edgetpu-L_tf
29 mobilenet_v3_small_1_0_tf2
30 resnet50 Caffe
31 resnet18
32 inception_v1
33 inception_v2
34 inception_v3
35 inception_v4
36 mobilenet_v2
37 squeezenet
38 resnet50_pt PyTorch
39 squeezenet_pt
40 inception_v3_pt
41 ofa_resnet50_0_9B_pt
42 person-orientation_pruned_558m_pt
43 ofa_depthwise_res50_pt
44 chen_color_resnet18_pt