提交 65989fe1 编写于 作者: M michaelowenliu

implement deeplabv3p with resnet50/101 and xception65

上级 2a026a69
......@@ -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
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册