跳转至

PaddleX模型列表(海光 DCU)

PaddleX 内置了多条产线,每条产线都包含了若干模块,每个模块包含若干模型,具体使用哪些模型,您可以根据下边的 benchmark 数据来选择。如您更考虑模型精度,请选择精度较高的模型,如您更考虑模型存储大小,请选择存储大小较小的模型。

图像分类模块

模型名称 Top1 Acc(%) 模型存储大小(M) 模型下载链接
ResNet18 71.0 41.5 M 推理模型/训练模型
ResNet34 74.6 77.3 M 推理模型/训练模型
ResNet50 76.5 90.8 M 推理模型/训练模型
ResNet101 77.6 158.7 M 推理模型/训练模型
ResNet152 78.3 214.2 M 推理模型/训练模型

注:以上精度指标为ImageNet-1k验证集 Top1 Acc。

语义分割模块

模型名称 mloU(%) 模型存储大小(M) 模型下载链接
Deeplabv3_Plus-R50 80.36 94.9 M 推理模型/训练模型
Deeplabv3_Plus-R101 81.10 162.5 M 推理模型/训练模型

注:以上精度指标为Cityscapes数据集 mloU。

评论