diff --git a/docs/source/experiments/segmentation.rst b/docs/source/experiments/segmentation.rst index 6e368a24..6d977037 100644 --- a/docs/source/experiments/segmentation.rst +++ b/docs/source/experiments/segmentation.rst @@ -55,6 +55,8 @@ Test Pre-trained Model .. _ENC50PC: https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/logs/encnet_resnet50_pcontext.log?raw=true .. _ENC101PC: https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/logs/encnet_resnet101_pcontext.log?raw=true .. _ENC50ADE: https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/logs/encnet_resnet50_ade.log?raw=true +.. _ENC101ADE: https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/logs/encnet_resnet101_ade.log?raw=true +.. _ENC101VOC: https://github.com/zhanghang1989/image-data/blob/master/encoding/segmentation/logs/encnet_resnet101_voc.log?raw=true .. raw:: html diff --git a/encoding/models/model_zoo.py b/encoding/models/model_zoo.py index 3870bbfd..135d410e 100644 --- a/encoding/models/model_zoo.py +++ b/encoding/models/model_zoo.py @@ -3,7 +3,6 @@ from .fcn import * from .psp import * from .encnet import * -from .deeplab import * __all__ = ['get_model'] @@ -33,7 +32,6 @@ def get_model(name, **kwargs): 'encnet_resnet101_ade': get_encnet_resnet101_ade, 'fcn_resnet50_ade': get_fcn_resnet50_ade, 'psp_resnet50_ade': get_psp_resnet50_ade, - 'deeplab_resnet50_ade': get_deeplab_resnet50_ade, } name = name.lower() if name not in models: