Unverified Commit abcee3c9 authored by Hang Zhang's avatar Hang Zhang Committed by GitHub
Browse files

patch (#125)

parent f8919197
...@@ -55,6 +55,8 @@ Test Pre-trained Model ...@@ -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 .. _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 .. _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 .. _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 .. raw:: html
......
...@@ -3,7 +3,6 @@ ...@@ -3,7 +3,6 @@
from .fcn import * from .fcn import *
from .psp import * from .psp import *
from .encnet import * from .encnet import *
from .deeplab import *
__all__ = ['get_model'] __all__ = ['get_model']
...@@ -33,7 +32,6 @@ def get_model(name, **kwargs): ...@@ -33,7 +32,6 @@ def get_model(name, **kwargs):
'encnet_resnet101_ade': get_encnet_resnet101_ade, 'encnet_resnet101_ade': get_encnet_resnet101_ade,
'fcn_resnet50_ade': get_fcn_resnet50_ade, 'fcn_resnet50_ade': get_fcn_resnet50_ade,
'psp_resnet50_ade': get_psp_resnet50_ade, 'psp_resnet50_ade': get_psp_resnet50_ade,
'deeplab_resnet50_ade': get_deeplab_resnet50_ade,
} }
name = name.lower() name = name.lower()
if name not in models: if name not in models:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment