From 481019bab06916a3359d323722eea089f7df909c Mon Sep 17 00:00:00 2001 From: Yi Zhu Date: Fri, 7 Aug 2020 12:35:33 -0700 Subject: [PATCH] Fastscnn improve (#1407) * add new fastscnn model * pip sanity check * change python --- docs/model_zoo/segmentation.rst | 41 +++++++++++++++++--------------- gluoncv/model_zoo/model_store.py | 2 +- tests/pylint.yml | 3 ++- 3 files changed, 25 insertions(+), 21 deletions(-) diff --git a/docs/model_zoo/segmentation.rst b/docs/model_zoo/segmentation.rst index b498728e64..25b3eede59 100644 --- a/docs/model_zoo/segmentation.rst +++ b/docs/model_zoo/segmentation.rst @@ -101,25 +101,25 @@ Pascal VOC Dataset Cityscapes Dataset ------------------ -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| Name | Method | pixAcc | mIoU | Command | log | -+=====================================+=================+===========+===========+=============================================================================================================================================+====================================================================================================================================+ -| psp_resnet101_citys | PSP [3]_ | 96.4 | 79.9 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| deeplab_resnet50_citys | DeepLabV3 [4]_ | 96.3 | 78.7 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| deeplab_resnet101_citys | DeepLabV3 [4]_ | 96.4 | 79.4 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| danet_resnet50_citys | DANet [7]_ | 96.3 | 78.5 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| danet_resnet101_citys | DANet [7]_ | 96.5 | 80.1 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| icnet_resnet50_citys | ICNet [5]_ | 95.5 | 74.5 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| fastscnn_citys | FastSCNN [8]_ | 94.3 | 67.2 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ -| deeplab_v3b_plus_wideresnet_citys | VPLR [6]_ | N/A | 83.5 | `shell script `_ | `log `_ | -+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| Name | Method | pixAcc | mIoU | Command | log | ++=====================================+===================+===========+===========+=============================================================================================================================================+====================================================================================================================================+ +| psp_resnet101_citys | PSP [3]_ | 96.4 | 79.9 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| deeplab_resnet50_citys | DeepLabV3 [4]_ | 96.3 | 78.7 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| deeplab_resnet101_citys | DeepLabV3 [4]_ | 96.4 | 79.4 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| danet_resnet50_citys | DANet [7]_ | 96.3 | 78.5 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| danet_resnet101_citys | DANet [7]_ | 96.5 | 80.1 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| icnet_resnet50_citys | ICNet [5]_ | 95.5 | 74.5 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| fastscnn_citys |FastSCNN [8]_ [10]_| 95.1 | 72.3 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ +| deeplab_v3b_plus_wideresnet_citys | VPLR [6]_ | N/A | 83.5 | `shell script `_ | `log `_ | ++-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+ MHP-V1 Dataset -------------- @@ -189,3 +189,6 @@ MS COCO .. [9] Hang Zhang, Chongruo Wu, Zhongyue Zhang, Yi Zhu, Zhi Zhang, Haibin Lin, Yue Sun, Tong He, Jonas Muller, R. Manmatha, Mu Li and Alex Smola \ "ResNeSt: Split-Attention Network" \ arXiv preprint (2020). +.. [10] Yi Zhu, Zhongyue Zhang, Chongruo Wu, Zhi Zhang, Tong He, Hang Zhang, R. Manmatha, Mu Li and Alexander Smola \ + "Improving Semantic Segmentation via Self-Training" \ + arXiv preprint arXiv:2004.14960 (2020). diff --git a/gluoncv/model_zoo/model_store.py b/gluoncv/model_zoo/model_store.py index ac87ac6d48..dd77e06bc9 100644 --- a/gluoncv/model_zoo/model_store.py +++ b/gluoncv/model_zoo/model_store.py @@ -126,7 +126,7 @@ ('63db8a7938586525256a0bdc6632ed986e4026cf', 'icnet_resnet50_mhpv1'), ('cf6a7bb3d55360933de647a8505f7936003902a4', 'deeplab_resnet50_citys'), ('eb8477a91efc244c85b364c0736664078aaf0e65', 'deeplab_resnet101_citys'), - ('b34e59e3e98e8d99f2609ff53f24629ed287b31c', 'fastscnn_citys'), + ('95aad0b699ae17c67caa44b3ead4b23474e98954', 'fastscnn_citys'), ('143e1f1c1c1f2d3a3416887e088ebfdd4e1e2345', 'danet_resnet50_citys'), ('6ead3d099f7a320846bddb51148c3fe3b5ade5c2', 'danet_resnet101_citys'), ('f5ece5ce1422eeca3ce2908004e469ffdf91fd41', 'yolo3_darknet53_voc'), diff --git a/tests/pylint.yml b/tests/pylint.yml index db381b0ae5..463b91567d 100644 --- a/tests/pylint.yml +++ b/tests/pylint.yml @@ -1,5 +1,6 @@ name: gluon_cv_pylint dependencies: -- python=3.6 +- python=3.7 +- pip=19.1.1 - pip: - pylint==2.4.4