支持模型列表及性能基准

Classification

networkfloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
resnet50_imagenet77.3776.9976.94ImageNet1x3x224x2240.5876540.9310.9041468.5422.652299.014
efficientnet_imagenet74.3174.2374.18ImageNet1x3x224x2240.33219776.520.414867.5340.924316.801
mixvargenet_imagenet71.3371.2371.04ImageNet1x3x224x2240.31320277.930.3665372.0380.875325.121
henet_tinye_imagenet77.6877.2276.92ImageNet1x3x224x2240.38111482.780.4983618.0331.0741631.96
henet_tinym_imagenet78.3877.9577.62ImageNet1x3x224x2240.39311256.830.5183380.9541.136290.305

Detection

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
fcos_efficientnetb3_mscocoefficientnetb348.0347.6547.58MS COCO1x3x896x8962.9221462.4014.049265.94913.88421.721
deform_detr_resnet50_mscocoresnet5044.3444.6544.80MS COCO1x3x800x133377.34729.456.336158.829--

Segmentation

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
unet_mobilenetv1_cityscapesMobileNetV168.0267.5667.53Cityscapes1x3x1024x20480.8775726.8621.1981146.0473.869331.277

3D检测

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
fcos3d_efficientnetb0_nuscenesefficientnetb030.6030.2730.31nuscenes1x3x512x8961.4613375.22.139562.2836.49138.403

Bev

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
bev_lss_efficientnetb0_multitask_nuscenesefficientnetb030.06(NDS) 20.62(MAP) 51.80(MeanIOU)30.10(NDS) 20.51(MAP) 51.78(MeanIOU)30.08(NDS) 20.46(MAP) 51.47(MeanIOU)nuscenes det && seg6x3x256x704, 10x128x128x2, 10x128x128x23.2541341.3714.597234.13320.688392.45
detr3d_efficientnetb3_nuscenesefficientnetb333.04(NDS) 27.52(MAP)32.84(NDS) 27.14(MAP)32.81(NDS) 27.06(MAP)nuscenes det6x3x512x140815.4735236.34523.06944.06175.859413.232
petr_efficientnetb3_nuscenesefficientnetb337.65(NDS) 30.38(MAP)37.26(NDS) 29.29(MAP)37.40(NDS) 29.33(MAP)nuscenes det6x3x512x140821.835180.75734.27929.532104.56879.593
bevformer_tiny_resnet50_detection_nuscenesresnet5037.12(NDS) 26.79(MAP)37.16(NDS) 26.50(MAP)37.15(NDS) 26.59(MAP)nuscenes det6x3x480x800, 1x2500x256, 1x50x50x2, 6x20x32x2, 1x100x50x2, 6x640x4x2, 1x2500x114.577270.03123.04344.148
bev_sparse_henet_tinym_nusceneshenet_tinym54.19(NDS) 43.38(MAP)52.23(NDS) 42.17(MAP)-nuscenes det6x3x256x704, 6x4x4, 1x384x11, 1x384x2568.217483.0711.81486.81829.91434.778
bev_sparse_det_maptr_flashocc_henet_tinym_nusceneshenet_tinym_imagenet52.34(det NDS) 41.33(det mAP) 59.58(map mAP) 31.89 (occ, MIOU)52.14(det NDS) 41.00(det mAP) 59.06(map mAP) 33.19 (occ, MIOU)-nuscenes det && map && occ3d6x3x256x704, 6x4x4, 1x384x11, 1x384x256, 10x128x128x2, 10x128x128x220.47163.2131.18532.51911.5765193.1247

Online Map Construction

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6E bpu latency (ms)J6E FPSJ6B bpu latency (ms)J6B FPS
maptroe_henet_tinym_bevformer_nusceneshenet_tinym_imagenet0.66320.65770.6567nuscenes6x3x480x800, 1x1x50x100, 6x20x100x2, 1x100x100x2, 6x2000x4x2, 1x5000x17.499521.6110.69696.24734.2829.894
maptroe_sparse_henet_tinym_nusceneshenet_tinym_imagenet0.59820.59990.5992nuscenes6x3x256x704, 6x4x49.19258.93611.55388.76729.22735.199

Occupancy Prediction

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6E bpu latency (ms)J6E FPSJ6B bpu latency (ms)J6B FPS
flashocc_henet_lss_occ3d_nusceneshenet_tinym_imagenet0.36740.36570.3693occ3d_nuscenes6x3x512x960, 10x128x128x2, 10x128x128x28.5437469.97710.58197.1229.33335.005

Multiple Object Track

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
motr_efficientnetb3_mot17efficientnetb358.0257.6257.76Mot171x3x800x1422, 1x256x2x128, 1x1x1x256, 1x4x2x1286.545604.4029.247111.78227.87236.806

Trajectory Prediction

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
qcnet_oe_argoverse2-80.0979.5478.21argoverse 2输入见下方list2.8421572.1233.888285.86812.659155.818
注解

qcnet_oe_argoverse2 的指标是 HitRate 这项。

qcnet_oe_argoverse2 模型输入shape为:

1x30x10, 1x10x30x30, 1x30x1, 1x1x30x1, 1x1x30x1, 1x1x30x1, 1x1x30x1, 1x1x30x80, 1x1x30x80, 1x1x30x80, 1x1x30x6, 1x1x30x6, 1x1x30x6, 1x1x30x6, 1x1x30x30, 1x1x30x30, 1x1x30x30, 1x30x2x128, 1x30x6x128, 1x80, 1x1x80x80, 1x1x80x80, 1x1x80x80, 1x1x80x50, 1x1x80x50, 1x1x80x50, 1x80x80, 1x80x50, 1x30x30, 1x30x1, 1x80x80

Lidar

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6M bpu latency (ms)J6M FPSJ6B bpu latency (ms)J6B FPS
pointpillars_kitti_carSequentialBottleNeck77.3176.8676.76KITTI3D150000x421.115327.56821.496197.2071652.4192.013
centerpoint_pointpillar_nuscenesSequentialBottleNeck58.32(NDS) 48.04(MAP)58.11(NDS) 47.85(MAP)58.14(NDS) 47.81(MAP)nuscenes det1x5x20x40000, 40000x47.957796.2329.483167.47338.27160.003

Lidar Fusion

networkbackbonefloatqatquantizationdatasetinput shapeJ6P bpu latency (ms)J6P FPSJ6E bpu latency (ms)J6E FPSJ6B bpu latency (ms)J6B FPS
bevfusion_pointpillar_henet_multisensor_multitask_nusceneshenet_tinym_imagenet64.28(NDS) 58.09(MAP) 51.77(MIOU)62.91(NDS) 57.48(MAP) 52.51(MIOU)-nuscenes det && occ3d1x5x20x40000, 40000x4, 6x3x512x960, 1x256x128x2, 6x5120x2x2, 1x16384x119.841233.4926.61843.57391.56415.304
bev_sparse_lidar_fusion_henet_tinym_nusceneshenet_tinym_imagenet66.64(NDS) 61.16(MAP)66.31(NDS), 60.70(mAP)65.96(NDS) 60.35(mAP)nuscenes det6x3x256x704, 6x4x4, 1x384x11, 1x384x256, 1x5x20x40000, 40000x417.314293.42923.37351.73985.03419.393