PaddleX 模型库官方介绍

本文包含模型的种类、预训练模型下载、推理速度和时间等

图像分类模型

表中模型准确率均为在ImageNet数据集上测试所得,表中符号-表示相关指标暂未测试,预测速度测试环境如下所示:

  • CPU的评估是在骁龙855(SD855)上完成。
  • GPU评估是在FP32+TensorRT配置下运行500次测得(去除前10次的warmup时间)。

 

移动端系列

ModelTop-1 AccTop-5 AccSD855 time(ms)
bs=1
Flops(G)Params(M)Model storage size(M)Download Address
MobileNetV1_
x0_25
0.51430.75463.219850.070.461.9Download link
MobileNetV1_
x0_5
0.63520.84739.5795990.281.315.2Download link
MobileNetV1_
x0_75
0.68810.882319.4363990.632.5510Download link
MobileNetV10.70990.896832.5230481.114.1916Download link
MobileNetV2_
x0_25
0.53210.76523.799250.051.56.1Download link
MobileNetV2_
x0_5
0.65030.85728.70210.171.937.8Download link
MobileNetV2_
x0_75
0.69830.890115.5313510.352.5810Download link
MobileNetV20.72150.906523.3176990.63.4414Download link
MobileNetV2_
x1_5
0.74120.916745.6238481.326.7626Download link
MobileNetV2_
x2_0
0.75230.925874.2916492.3211.1343Download link
MobileNetV3_
large_x1_25
0.76410.929528.2177010.7147.4429Download link
MobileNetV3_
large_x1_0
0.75320.923119.308350.455.4721Download link
MobileNetV3_
large_x0_75
0.73140.910813.56460.2963.9116Download link
MobileNetV3_
large_x0_5
0.69240.88527.493150.1382.6711Download link
MobileNetV3_
large_x0_35
0.64320.85465.136950.0772.18.6Download link
MobileNetV3_
small_x1_25
0.70670.89519.27450.1953.6214Download link
MobileNetV3_
small_x1_0
0.68240.88066.54630.1232.9412Download link
MobileNetV3_
small_x0_75
0.66020.86335.284350.0882.379.6Download link
MobileNetV3_
small_x0_5
0.59210.81523.351650.0431.97.8Download link
MobileNetV3_
small_x0_35
0.53030.76372.63520.0261.666.9Download link
MobileNetV3_
small_x0_35_ssld
0.55550.77712.63520.0261.666.9Download link
MobileNetV3_
large_x1_0_ssld
0.78960.944819.308350.455.4721Download link
MobileNetV3_small_
x1_0_ssld
0.71290.90106.54630.1232.9412Download link
ShuffleNetV20.68800.884510.9410.282.269Download link
ShuffleNetV2_
x0_25
0.49900.73792.3290.030.62.7Download link
ShuffleNetV2_
x0_33
0.53730.77052.643350.040.642.8Download link
ShuffleNetV2_
x0_5
0.60320.82264.26130.081.365.6Download link
ShuffleNetV2_
x1_5
0.71630.901519.35220.583.4714Download link
ShuffleNetV2_
x2_0
0.73150.912034.7701491.127.3228Download link
ShuffleNetV2_
swish
0.70030.891716.0231510.292.269.1Download link

其他系列

ModelTop-1 AccTop-5 Acctime(ms)
bs=1
time(ms)
bs=4
Flops(G)Params(M)Download Address
ResNet180.70980.89921.456063.563053.6611.69Download link
ResNet18_vd0.72260.90801.545573.853634.1411.71Download link
ResNet340.74570.92142.349575.898217.3621.8Download link
ResNet34_vd0.75980.92982.434276.222577.3921.82Download link
ResNet500.76500.93003.477127.844218.1925.56Download link
ResNet50_vd0.79120.94443.531318.090578.6725.58Download link
ResNet1010.77560.93646.0712513.4057315.5244.55Download link
ResNet101_vd0.80170.94976.1170413.7622216.144.57Download link
ResNet1520.78260.93968.5019819.1707323.0560.19Download link
ResNet152_vd0.80590.95308.5437619.5215723.5360.21Download link
ResNet200_vd0.80930.953310.8061925.0173130.5374.74Download link
ResNet50_vd_
ssld
0.82390.96103.531318.090578.6725.58Download link
ResNet101_vd_
ssld
0.83730.96696.1170413.7622216.144.57Download link
AlexNet0.5670.7921.449932.466961.37061.090Download link
DarkNet530.7800.9414.1082912.171418.58041.600Download link
DenseNet1210.75660.92584.404479.326235.697.98Download link
DenseNet1610.78570.941410.3915222.1555515.4928.68Download link
DenseNet1690.76810.93316.4359812.988326.7414.15Download link
DenseNet2010.77630.93668.2065217.458388.6120.01Download link
DenseNet2640.77960.938512.1472226.2770711.5433.37Download link
HRNet_W18_C0.76920.93397.4063613.297524.1421.29Download link
HRNet_W30_C0.78040.94029.5759417.3548516.2337.71Download link
HRNet_W32_C0.78280.94249.4980717.7292117.8641.23Download link
HRNet_W40_C0.78770.944712.1220225.6818425.4157.55Download link
HRNet_W44_C0.79000.945113.1985832.2520229.7967.06Download link
HRNet_W48_C0.78950.944213.7076134.4357234.5877.47Download link
HRNet_W64_C0.79300.946117.5752747.953357.83128.06Download link
Xception410.79300.94534.9693917.0136116.7422.69Download link
Xception650.81000.95497.2615825.8877825.9535.48Download link
Xception710.81110.95458.7245731.5554931.7737.28Download link

目标检测模型

  • 所有模型均在COCO17数据集中训练和测试。
  • 除非特殊说明,所有ResNet骨干网络采用ResNet-B结构。
  • 推理时间(fps): 推理时间是在一张Tesla V100的GPU上测试所有验证集得到,单位是fps(图片数/秒), cuDNN版本是7.5,包括数据加载、网络前向执行和后处理, batch size是1。

Faster RCNN on COCO

骨架网络网络类型推理时间(fps)Box AP下载
ResNet50Faster----36.7下载链接
ResNet50-vdFaster----37.6下载链接
ResNet101Faster----39.0下载链接
ResNet34-FPNFaster----37.8下载链接
ResNet34-vd-FPNFaster----38.5下载链接
ResNet50-FPNFaster----38.4下载链接
ResNet50-vd-FPNFaster----39.5下载链接
ResNet101-vd-FPNFaster----42.0下载链接
ResNeXt101-vd-FPNFaster----43.4下载链接
ResNet50-vd-SSLDv2-FPNFaster----41.4下载链接

YOLOv3 on COCO

骨架网络输入尺寸推理时间(fps)Box AP下载
DarkNet53608----39.0下载链接
ResNet50_vd608----39.1下载链接
ResNet34608----36.2下载链接
MobileNet-V1608----29.4下载链接
MobileNet-V3608----31.4下载链接
MobileNet-V1-SSLD608----31.0下载链接

YOLOv3 on Pasacl VOC

骨架网络输入尺寸推理时间(fps)Box AP下载
MobileNet-V1608-75.2下载链接
MobileNet-V3608-79.6下载链接
MobileNet-V1-SSLD608-78.3下载链接
MobileNet-V3-SSLD608-80.4下载链接

PP-YOLO on COCO

模型骨干网络输入尺寸Box APvalBox APtestV100 FP32(FPS)V100 TensorRT FP16(FPS)模型下载
PP-YOLOResNet50vd60844.845.272.9155.6model
PP-YOLOv2ResNet50vd64049.149.568.9106.5model
PP-YOLOv2ResNet101vd64049.750.349.587.0model

注意:

  • PP-YOLO模型推理速度测试采用单卡V100,batch size=1进行测试,使用CUDA 10.2, CUDNN 7.5.1,TensorRT推理速度测试使用TensorRT 5.1.2.2。
  • PP-YOLO模型FP32的推理速度是使用Paddle预测库进行推理速度benchmark测试结果, 且测试的均为不包含数据预处理和模型输出后处理(NMS)的数据(与YOLOv4(AlexyAB)测试方法一致)。
  • TensorRT FP16的速度测试相比于FP32去除了yolo_box(bbox解码)部分耗时,即不包含数据预处理,bbox解码和NMS(与YOLOv4(AlexyAB)测试方法一致)。

PP-YOLO on Pascal VOC

模型骨干网络输入尺寸Box AP50val模型下载
PP-YOLOResNet50vd60884.9model

PP-YOLO tiny on COCO

模型模型体积后量化模型体积输入尺寸Box APvalKirin 990 1xCore (FPS)模型下载量化后模型
PP-YOLO tiny4.2MB1.3M41622.765.4model预测模型
  • PP-YOLO-tiny 模型推理速度测试环境配置为麒麟990芯片4线程,arm8架构。
  • 我们也提供的PP-YOLO-tiny的后量化压缩模型,将模型体积压缩到1.3M,对精度和预测速度基本无影响

实例分割模型

  • 所有模型均在COCO17数据集中训练和测试。
  • 除非特殊说明,所有ResNet骨干网络采用ResNet-B结构。
  • 推理时间(fps): 推理时间是在一张Tesla V100的GPU上测试所有验证集得到,单位是fps(图片数/秒), cuDNN版本是7.5,包括数据加载、网络前向执行和后处理, batch size是1。
骨架网络网络类型推理时间(fps)Box APMask AP下载
ResNet50Mask----37.432.8下载链接
ResNet50Mask----39.734.5下载链接
ResNet50-FPNMask----39.235.6下载链接
ResNet50-FPNMask----40.536.7下载链接
ResNet50-vd-FPNMask----40.336.4下载链接
ResNet50-vd-FPNMask----41.437.5下载链接
ResNet101-FPNMask----40.636.6下载链接
ResNet101-vd-FPNMask----42.438.1下载链接
ResNeXt101-vd-FPNMask----44.039.5下载链接
ResNeXt101-vd-FPNMask----44.639.8下载链接
ResNet50-vd-SSLDv2-FPNMask----42.038.2下载链接
ResNet50-vd-SSLDv2-FPNMask----42.738.9下载链接

语义分割模型

以下指标均在Pascal VOC验证集上测试得到,表中符号-表示相关指标暂未测试。

ModelBackboneResolutionmIoULinks
DeepLabV3PResNet50_vd512x51280.66%model
DeepLabV3PResNet101_vd512x51280.60%model

以下指标均在Cityscapes验证集上测试得到,表中符号-表示相关指标暂未测试。

ModelBackboneResolutionmIoULinks
UNet-1024x51265.00%model
DeepLabV3PResNet50_vd1024x51280.36%model
DeepLabV3PResNet101_vd1024x51281.10%model
Fast SCNN-1024x102469.31%model
HRNet_W18-1024x51278.97%model
HRNet_W48-1024x51280.70%model
BiSeNetv2-1024x102473.19%model
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

武大人民泌外I科人工智能团队

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值