车辆分类 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

“Vehicle Classification”(车辆分类)库用于对车辆图像(车辆品牌或车辆类型)进行分类。此类神经网络是在 CompCars 上训练的,可以识别来自 163 个类或 12 个类的对象。Vitis AI Library 将 vehicle_make_resnet18_ptvehicle_type_resnet18_pt 集成到赛灵思库中。输入是一张包含一个对象的图片,输出则是前 K 个可能性最高的类别。

图 1. 车辆分类示例

表 1. 车辆分类模型
编号 模型名称 框架
1 vehicle_type_resnet18_pt PyTorch
2 vehicle_make_resnet18_pt PyTorch