Add Deeplab resnest200 ade pretrained weights (#1456)

* add pretrained weights deeplab resnest200 ade * add unittest * add numbers
dmlc · Sep 29, 2020 · 9eeeb17 · 9eeeb17
1 parent ea22187
commit 9eeeb17
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 0 deletions.
diff --git a/docs/model_zoo/segmentation.rst b/docs/model_zoo/segmentation.rst
@@ -61,6 +61,8 @@ ADE20K Dataset
 +-----------------------+--------------------------------+-----------+-----------+------------------------------------------------------------------------------------------------------------------------------+---------------------------------------------------------------------------------------------------------------------+
 | deeplab_resnest101_ade| DeepLabV3 + ResNeSt [4]_ [9]_  | 82.1      | 46.9      | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnest101_ade.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnest101_ade.log>`_|
 +-----------------------+--------------------------------+-----------+-----------+------------------------------------------------------------------------------------------------------------------------------+---------------------------------------------------------------------------------------------------------------------+
+| deeplab_resnest200_ade| DeepLabV3 + ResNeSt [4]_ [9]_  | 82.5      | 48.4      |                                                                                                                              |                                                                                                                     |
++-----------------------+--------------------------------+-----------+-----------+------------------------------------------------------------------------------------------------------------------------------+---------------------------------------------------------------------------------------------------------------------+
 | deeplab_resnest269_ade| DeepLabV3 + ResNeSt [4]_ [9]_  | 82.6      | 47.6      | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnest269_ade.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnest269_ade.log>`_|
 +-----------------------+--------------------------------+-----------+-----------+------------------------------------------------------------------------------------------------------------------------------+---------------------------------------------------------------------------------------------------------------------+
 

diff --git a/gluoncv/model_zoo/model_store.py b/gluoncv/model_zoo/model_store.py
@@ -116,6 +116,7 @@
     ('bf1584dfcec12063eff3075ee643e181c0f6d443', 'deeplab_resnet101_ade'),
     ('a8312db6e30a464151580f2bda83479786455724', 'deeplab_resnest50_ade'),
     ('6d05c630fb7acb38615f7f4d360fb90f47b25042', 'deeplab_resnest101_ade'),
+    ('9134a6f7d1399b549d5803d8faed3dfc74efc0d8', 'deeplab_resnest200_ade'),
     ('d45b33dedf4cca13b8147213c5360e30f93685bd', 'deeplab_resnest269_ade'),
     ('09f89cad0e107cb2bffdb1b07706ba31798096f2', 'psp_resnet101_coco'),
     ('2c2f4e1c2b11461b52598a4b2038bccbcfc166eb', 'psp_resnet101_voc'),

diff --git a/tests/model_zoo/test_model_zoo.py b/tests/model_zoo/test_model_zoo.py
@@ -421,6 +421,7 @@ def test_segmentation_models():
               'fcn_resnet101_voc', 'psp_resnet101_voc', 'deeplab_resnet101_voc',
               'fcn_resnet50_ade', 'psp_resnet50_ade', 'deeplab_resnet50_ade',
               'fcn_resnet101_ade', 'psp_resnet101_ade', 'deeplab_resnet101_ade',
+              'deeplab_resnest200_ade',
               'psp_resnet101_citys', 'deeplab_resnet152_voc', 'deeplab_resnet152_coco',
               'deeplab_v3b_plus_wideresnet_citys', 'icnet_resnet50_citys',
               'deeplab_resnet50_citys', 'deeplab_resnet101_citys']