跳到主要内容

模型收益

Quant_INT8-8量化收益

以resnet-18为例,使用轻量化工具后(Quant_INT8-8量化)的收益如下。

框架数据集模型原始精度重训练后精度非量化OM离线 模型体积(MB)Quant_INT8-8量化OM 离线模型体积(MB)
TensorFlowImageNetresnet-18(v2)70.0%70.0%22.45711.9

网络结构搜索工具分类场景收益对比

以resnet-18为例,使用轻量化工具后(网络结构搜索)的收益如下。

框架数据集模型参数量(M)精度
TensorFlowImageNetResNet-1811.6970.32%
TensorFlowImageNetNASEA10.9372.13%[1]
PyTorchImageNetResNet-1811.6969.6%
PyTorchImageNetNASEA11.472.88%

检测场景收益对比

框架数据集模型计算量(G)mAP@[.5, .95][2]
SSD(backbone:ResNet-18)COCOResNet-1811.617.8[3]
SSD(backbone:ResNet-18)COCONASEA9.2518.4[3]

分割场景收益对比

框架数据集模型计算量(G)mIOU[4]
TensorFlowVOCResNet-18 + Deeplab v340.954.1[5]
TensorFlowVOCNASEA28.356.1[5]
PyTorchVOCResNet-18 + Deeplab v343.764.2
PyTorchVOCNASEA30.765.1

[1] 此精度是使用tensorpack重训练模型得出。

[2] 此指标的计算方法为:IoU(Intersection over Union)从0.5~0.95区间上,以0.05为间隔计算AP的值,再计算所有AP的均值。

[3] 此精度是在COCO val2017数据集上测试得出。

[4] 此指标为平均交并比,计算方法为先求每个类别的交并比,再平均。

[5] 此精度是在VOC val2012数据集上测试得出。