Skip to content

P
PaddleSeg

PaddlePaddle / PaddleSeg

通知 285
Star 8
Fork 1
P
PaddleSeg
未验证 提交 e39762b8 编写于 作者: W wuzewu 提交者： GitHub
浏览文件
操作

Merge pull request #344 from michaelowenliu/develop

上级 49176dc0 65989fe1
隐藏空白更改
内联 并排
Showing with 8 addition and 1 deletion
dygraph/models/__init__.py
浏览文件 @ e39762b8
...@@ -14,6 +14,7 @@ ...@@ -14,6 +14,7 @@
from .unet import UNet from .unet import UNet
from .hrnet import * from .hrnet import *
from .deeplab import *
MODELS = { MODELS = {
"UNet": UNet, "UNet": UNet,
...@@ -36,5 +37,11 @@ MODELS = { ...@@ -36,5 +37,11 @@ MODELS = {
"SE_HRNet_W44": SE_HRNet_W44, "SE_HRNet_W44": SE_HRNet_W44,
"SE_HRNet_W48": SE_HRNet_W48, "SE_HRNet_W48": SE_HRNet_W48,
"SE_HRNet_W60": SE_HRNet_W60, "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
} }
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或