“Vehicle Classification”(车辆分类)库用于对车辆图像(车辆品牌或车辆类型)进行分类。此类神经网络是在 CompCars 上训练的,可以识别来自 163 个类或 12 个类的对象。Vitis AI Library 将 vehicle_make_resnet18_pt 和 vehicle_type_resnet18_pt 集成到赛灵思库中。输入是一张包含一个对象的图片,输出则是前 K 个可能性最高的类别。
图 1. 车辆分类示例
编号 | 模型名称 | 框架 |
---|---|---|
1 | vehicle_type_resnet18_pt | PyTorch |
2 | vehicle_make_resnet18_pt | PyTorch |