diff --git a/dygraph/models/__init__.py b/dygraph/models/__init__.py index 0057d88f154f93542c00b0d3dd22a8e52005764c..62af2a91fa22edd69ed9aeae9de33958bd810959 100644 --- a/dygraph/models/__init__.py +++ b/dygraph/models/__init__.py @@ -14,6 +14,7 @@ from .unet import UNet from .hrnet import * +from .deeplab import * MODELS = { "UNet": UNet, @@ -36,5 +37,11 @@ MODELS = { "SE_HRNet_W44": SE_HRNet_W44, "SE_HRNet_W48": SE_HRNet_W48, "SE_HRNet_W60": SE_HRNet_W60, - "SE_HRNet_W64": SE_HRNet_W64 + "SE_HRNet_W64": SE_HRNet_W64, + "DeepLabV3P": DeepLabV3P, + "deeplabv3p_resnet101_vd": deeplabv3p_resnet101_vd, + "deeplabv3p_resnet101_vd_os8": deeplabv3p_resnet101_vd_os8, + "deeplabv3p_resnet50_vd": deeplabv3p_resnet50_vd, + "deeplabv3p_resnet50_vd_os8": deeplabv3p_resnet50_vd_os8, + "deeplabv3p_xception65_deeplab": deeplabv3p_xception65_deeplab }