Merge pull request #344 from michaelowenliu/develop

e39762b8 · wuzewu · GitHub · 49176dc0 · 65989fe1 · e39762b8
隐藏空白更改
内联并排

Showing with 8 addition and 1 deletion

dygraph/models/__init__.py dygraph/models/__init__.py +8 -1

未找到文件。
--- 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
 }