mmsegmentation/configs/pspnet
Jerry Jiarui XU f3cf96b63e [Bug fix] Fixed ADE20k test (#359)
* [Bug fix] Fixed ADE20k test

* fixed ade cfg
2021-01-24 02:17:59 -08:00
..
README.md [Improvement] Add tags for each README.md (#340) 2021-01-10 21:35:09 -08:00
pspnet_r18-d8_512x1024_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r18-d8_769x769_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r18b-d8_512x1024_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r18b-d8_769x769_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r50-d8_480x480_40k_pascal_context.py Add Pascal Context to mmsegmentation (#133) 2020-09-22 14:56:13 +08:00
pspnet_r50-d8_480x480_80k_pascal_context.py Add Pascal Context to mmsegmentation (#133) 2020-09-22 14:56:13 +08:00
pspnet_r50-d8_512x512_20k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r50-d8_512x512_40k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r50-d8_512x512_80k_ade20k.py [Bug fix] Fixed ADE20k test (#359) 2021-01-24 02:17:59 -08:00
pspnet_r50-d8_512x512_160k_ade20k.py [Bug fix] Fixed ADE20k test (#359) 2021-01-24 02:17:59 -08:00
pspnet_r50-d8_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r50-d8_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r50-d8_769x769_40k_cityscapes.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
pspnet_r50-d8_769x769_80k_cityscapes.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
pspnet_r50b-d8_512x1024_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r50b-d8_769x769_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r101-d8_480x480_40k_pascal_context.py Add Pascal Context to mmsegmentation (#133) 2020-09-22 14:56:13 +08:00
pspnet_r101-d8_480x480_80k_pascal_context.py Add Pascal Context to mmsegmentation (#133) 2020-09-22 14:56:13 +08:00
pspnet_r101-d8_512x512_20k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_512x512_40k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_512x512_80k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_512x512_160k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_769x769_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101-d8_769x769_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
pspnet_r101b-d8_512x1024_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00
pspnet_r101b-d8_769x769_80k_cityscapes.py Add more models (#316) 2020-12-29 17:56:06 -08:00

README.md

Pyramid Scene Parsing Network

Introduction

[ALGORITHM]

@inproceedings{zhao2017pspnet,
  title={Pyramid Scene Parsing Network},
  author={Zhao, Hengshuang and Shi, Jianping and Qi, Xiaojuan and Wang, Xiaogang and Jia, Jiaya},
  booktitle={CVPR},
  year={2017}
}

Results and models

Cityscapes

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) download
PSPNet R-50-D8 512x1024 40000 6.1 4.07 77.85 79.18 model | log
PSPNet R-101-D8 512x1024 40000 9.6 2.68 78.34 79.74 model | log
PSPNet R-50-D8 769x769 40000 6.9 1.76 78.26 79.88 model | log
PSPNet R-101-D8 769x769 40000 10.9 1.15 79.08 80.28 model | log
PSPNet R-18-D8 512x1024 80000 1.7 15.71 74.87 76.04 model | log
PSPNet R-50-D8 512x1024 80000 - - 78.55 79.79 model | log
PSPNet R-101-D8 512x1024 80000 - - 79.76 81.01 model | log
PSPNet R-18-D8 769x769 80000 1.9 6.20 75.90 77.86 model | log
PSPNet R-50-D8 769x769 80000 - - 79.59 80.69 model | log
PSPNet R-101-D8 769x769 80000 - - 79.77 81.06 model | log
PSPNet R-18b-D8 512x1024 80000 1.5 16.28 74.23 75.79 model | log
PSPNet R-50b-D8 512x1024 80000 6.0 4.30 78.22 79.46 model | log
PSPNet R-101b-D8 512x1024 80000 9.5 2.76 79.69 80.79 model | log
PSPNet R-18b-D8 769x769 80000 1.7 6.41 74.92 76.90 model | log
PSPNet R-50b-D8 769x769 80000 6.8 1.88 78.50 79.96 model | log
PSPNet R-101b-D8 769x769 80000 10.8 1.17 78.87 80.04 model | log

ADE20K

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) download
PSPNet R-50-D8 512x512 80000 8.5 23.53 41.13 41.94 model | log
PSPNet R-101-D8 512x512 80000 12 15.30 43.57 44.35 model | log
PSPNet R-50-D8 512x512 160000 - - 42.48 43.44 model | log
PSPNet R-101-D8 512x512 160000 - - 44.39 45.35 model | log

Pascal VOC 2012 + Aug

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) download
PSPNet R-50-D8 512x512 20000 6.1 23.59 76.78 77.61 model | log
PSPNet R-101-D8 512x512 20000 9.6 15.02 78.47 79.25 model | log
PSPNet R-50-D8 512x512 40000 - - 77.29 78.48 model | log
PSPNet R-101-D8 512x512 40000 - - 78.52 79.57 model | log

Pascal Context

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) download
PSPNet R-101-D8 480x480 40000 8.8 9.68 46.60 47.78 model | log
PSPNet R-101-D8 480x480 80000 - - 46.03 47.15 model | log