Skip to content

Latest commit

 

History

History
44 lines (32 loc) · 3.25 KB

model_zoo.md

File metadata and controls

44 lines (32 loc) · 3.25 KB

PaddleSeg 预训练模型

PaddleSeg对所有内置的分割模型都提供了公开数据集下的预训练模型,通过加载预训练模型后训练可以在自定义数据集中得到更稳定地效果。

ImageNet预训练模型

所有Imagenet预训练模型来自于PaddlePaddle图像分类库,想获取更多细节请点击这里

模型 数据集合 Depth multiplier 下载地址 Accuray Top1/5 Error
MobileNetV2_1.0x ImageNet 1.0x MobileNetV2_1.0x 72.15%/90.65%
MobileNetV2_0.25x ImageNet 0.25x MobileNetV2_0.25x 53.21%/76.52%
MobileNetV2_0.5x ImageNet 0.5x MobileNetV2_0.5x 65.03%/85.72%
MobileNetV2_1.5x ImageNet 1.5x MobileNetV2_1.5x 74.12%/91.67%
MobileNetV2_2.0x ImageNet 2.0x MobileNetV2_2.0x 75.23%/92.58%

用户可以结合实际场景的精度和预测性能要求,选取不同Depth multiplier参数的MobileNet模型。

模型 数据集合 下载地址 Accuray Top1/5 Error
Xception41 ImageNet Xception41_pretrained.tgz 79.5%/94.38%
Xception65 ImageNet Xception65_pretrained.tgz 80.32%/94.47%
Xception71 ImageNet coming soon --

COCO预训练模型

数据集为COCO实例分割数据集合转换成的语义分割数据集合

模型 数据集合 下载地址 Output Strid multi-scale test mIoU
DeepLabv3+/MobileNetv2/bn COCO deeplab_mobilenet_x1_0_coco.tgz 16 -- --
DeeplabV3+/Xception65/bn COCO xception65_coco.tgz 16 -- --
U-Net/bn COCO unet_coco.tgz 16 -- --

Cityscapes预训练模型

train数据集合为Cityscapes训练集合,测试为Cityscapes的验证集合

模型 数据集合 下载地址 Output Stride mutli-scale test mIoU on val
DeepLabv3+/MobileNetv2/bn Cityscapes mobilenet_cityscapes.tgz 16 false 0.698
DeepLabv3+/Xception65/gn Cityscapes deeplabv3p_xception65_gn_cityscapes.tgz 16 false 0.7824
DeepLabv3+/Xception65/bn Cityscapes deeplabv3p_xception65_bn_cityscapes_.tgz 16 false 0.7930
ICNet/bn Cityscapes icnet_cityscapes.tgz 16 false 0.6831