分类 - 3.5 简体中文

“Classification”（分类）库用于对图像进行分类。此类神经网络是在 ImageNet 数据集的 1000 类 ILSVRC 子集上训练的，可以预测图像中的对象属于每个类成员的概率（按类）。Vitis AI Library 支持各种分类网络，包括但不限于 ResNet18、ResNet50、Inception_v1、Inception_v2、Inception_v3、Inception_v4、VGG、mobilenet_v1、mobilenet_v2 和 Squeezenet。输入是一张包含一个对象的图像，输出则是前 K 个可能性最高的类别。

下表列出了 Vitis AI Library 支持的分类模型。