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
.. _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
......
......@@ -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:
......
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