PointPillars 是高效网络,用于在点云上进行实时三维对象检测。此模型是基于 nuScenes 数据集来训练的,可在环视范围内为 10 个类(包括某些类型的车辆、行人、障碍物和交通锥)提供三维边界框和速度预测。PointPillars 使用多次扫描点云作为输入,可以实现更高准确度的三维对象检测和速度估算,但代价是会增加预处理部分的复杂性。
图 1. PointPillars_nuscenes 示例
下表列出了 Vitis AI Library 支持的 PointPillars_nuscenes 模型。
编号 | 模型名称 | 框架 |
---|---|---|
1 | pointpillars_nuscenes_40000_64_0_pt | PyTorch |
2 | pointpillars_nuscenes_40000_64_1_pt | PyTorch |