Skip to content

Commit

Permalink
Fastscnn improve (#1407)
Browse files Browse the repository at this point in the history 
* add new fastscnn model

* pip sanity check

* change python
  • Loading branch information
@bryanyzhu
bryanyzhu authored Aug 7, 2020
1 parent c73116d commit 481019b
Show file tree
Hide file tree
Showing 3 changed files with 25 additions and 21 deletions.
41 changes: 22 additions & 19 deletions docs/model_zoo/segmentation.rst
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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 <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/psp_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/psp_resnet101_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_resnet50_citys | DeepLabV3 [4]_ | 96.3 | 78.7 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet50_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_resnet101_citys | DeepLabV3 [4]_ | 96.4 | 79.4 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet101_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| danet_resnet50_citys | DANet [7]_ | 96.3 | 78.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet50_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| danet_resnet101_citys | DANet [7]_ | 96.5 | 80.1 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet101_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| icnet_resnet50_citys | ICNet [5]_ | 95.5 | 74.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| fastscnn_citys | FastSCNN [8]_ | 94.3 | 67.2 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/fastscnn_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/fastscnn_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_v3b_plus_wideresnet_citys | VPLR [6]_ | N/A | 83.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_v3b_plus_wideresnet_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_v3b_plus_wideresnet_citys.log>`_ |
+-------------------------------------+-----------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| Name | Method | pixAcc | mIoU | Command | log |
+=====================================+===================+===========+===========+=============================================================================================================================================+====================================================================================================================================+
| psp_resnet101_citys | PSP [3]_ | 96.4 | 79.9 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/psp_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/psp_resnet101_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_resnet50_citys | DeepLabV3 [4]_ | 96.3 | 78.7 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet50_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_resnet101_citys | DeepLabV3 [4]_ | 96.4 | 79.4 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_resnet101_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| danet_resnet50_citys | DANet [7]_ | 96.3 | 78.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet50_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| danet_resnet101_citys | DANet [7]_ | 96.5 | 80.1 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet101_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/danet_resnet101_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| icnet_resnet50_citys | ICNet [5]_ | 95.5 | 74.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| fastscnn_citys |FastSCNN [8]_ [10]_| 95.1 | 72.3 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/fastscnn_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/fastscnn_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+
| deeplab_v3b_plus_wideresnet_citys | VPLR [6]_ | N/A | 83.5 | `shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_v3b_plus_wideresnet_citys.sh>`_ | `log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/deeplab_v3b_plus_wideresnet_citys.log>`_ |
+-------------------------------------+-------------------+-----------+-----------+---------------------------------------------------------------------------------------------------------------------------------------------+------------------------------------------------------------------------------------------------------------------------------------+

MHP-V1 Dataset
--------------
Expand Down Expand Up @@ -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).
2 changes: 1 addition & 1 deletion gluoncv/model_zoo/model_store.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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'),
Expand Down
3 changes: 2 additions & 1 deletion tests/pylint.yml
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
name: gluon_cv_pylint
dependencies:
- python=3.6
- python=3.7
- pip=19.1.1
- pip:
- pylint==2.4.4

0 comments on commit 481019b

Please sign in to comment.