目前在github上只能找到2个项目
-
- tkDNN
两者都是使用CenterNet,但第1个基于TensorRT5,无法与当前最新的TensorRT6和TensorRT7兼容。经测试,第1个无法在Xavier+Jetpack 4.3/4.4上部署,因此选择部署第二个tkDNN。
- tkDNN
1. 基本信息
- README
- 模型网络:Centernet3D (DLA34 backend)
- 训练代码: CenterNet
- 输入尺寸: 512x512
- 当前类别数: 3类,“person”, “car”, “bicycle”
2. 精度
TRT未报告精度。
原torch测试精度为
其中E、M、H表示easy、moderate 、hard难度。
AP:平均精度
AOS:检测目标旋转角度的准确率
BEV:BEV视角下检测框的准确率