Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 285
Star 8
Fork 1
P
PaddleSeg
提交 05996d63 编写于 作者: M michaelowenliu
浏览文件
操作

add dim_reduction to PPModule

上级 c619f4ee
隐藏空白更改
内联 并排
Showing with 4 addition and 4 deletion
dygraph/models/pspnet.py
浏览文件 @ 05996d63
...@@ -209,23 +209,23 @@ class PPModule(fluid.dygraph.Layer): ...@@ -209,23 +209,23 @@ class PPModule(fluid.dygraph.Layer):
@manager.MODELS.add_component @manager.MODELS.add_component
def pspnet_resnet101_vd(*args, **kwargs): def pspnet_resnet101_vd(*args, **kwargs):
pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet101_vd_ssld_imagenet" pretrained_model = None
return PSPNet(backbone='ResNet101_vd', pretrained_model=pretrained_model, **kwargs) return PSPNet(backbone='ResNet101_vd', pretrained_model=pretrained_model, **kwargs)
@manager.MODELS.add_component @manager.MODELS.add_component
def pspnet_resnet101_vd_os8(*args, **kwargs): def pspnet_resnet101_vd_os8(*args, **kwargs):
pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet101_vd_ssld_imagenet" pretrained_model = None
return PSPNet(backbone='ResNet101_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs) return PSPNet(backbone='ResNet101_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs)
@manager.MODELS.add_component @manager.MODELS.add_component
def pspnet_resnet50_vd(*args, **kwargs): def pspnet_resnet50_vd(*args, **kwargs):
pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet50_vd_ssld_v2_imagenet" pretrained_model = None
return PSPNet(backbone='ResNet50_vd', pretrained_model=pretrained_model, **kwargs) return PSPNet(backbone='ResNet50_vd', pretrained_model=pretrained_model, **kwargs)
@manager.MODELS.add_component @manager.MODELS.add_component
def pspnet_resnet50_vd_os8(*args, **kwargs): def pspnet_resnet50_vd_os8(*args, **kwargs):
pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet50_vd_ssld_v2_imagenet" pretrained_model = None
return PSPNet(backbone='ResNet50_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs) return PSPNet(backbone='ResNet50_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs)
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或