mmsegmentation/configs/psanet
谢昕辰 725d5aa002 [Feature] support mim (#549)
* dice loss

* format code, add docstring and calculate denominator without valid_mask

* minor change

* restore

* add metafile

* add manifest.in and add config at setup.py

* add requirements

* modify manifest

* modify manifest

* Update MANIFEST.in

* add metafile

* add metadata

* fix typo

* Update metafile.yml

* Update metafile.yml

* minor change

* Update metafile.yml

* add subfix

* fix mmshow

* add more  metafile

* add config to model_zoo

* fix bug

* Update mminstall.txt

* [fix] Add models

* [Fix] Add collections

* [fix] Modify collection name

* [Fix] Set datasets to unet metafile

* [Fix] Modify collection names

* complement inference time
2021-05-31 15:07:24 -07:00
..
README.md comment tag (#505) 2021-04-24 09:58:59 -07:00
metafile.yml [Feature] support mim (#549) 2021-05-31 15:07:24 -07:00
psanet_r50-d8_512x512_20k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
psanet_r50-d8_512x512_40k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
psanet_r50-d8_512x512_80k_ade20k.py [Bug fix] Fixed ADE20k test (#359) 2021-01-24 02:17:59 -08:00
psanet_r50-d8_512x512_160k_ade20k.py [Bug fix] Fixed ADE20k test (#359) 2021-01-24 02:17:59 -08:00
psanet_r50-d8_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
psanet_r50-d8_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
psanet_r50-d8_769x769_40k_cityscapes.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
psanet_r50-d8_769x769_80k_cityscapes.py [Improvement] Move train_cfg/test_cfg inside model (#341) 2021-01-19 17:06:23 -08:00
psanet_r101-d8_512x512_20k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_512x512_40k_voc12aug.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_512x512_80k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_512x512_160k_ade20k.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_512x1024_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_512x1024_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_769x769_40k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00
psanet_r101-d8_769x769_80k_cityscapes.py init commit 2020-07-10 02:39:01 +08:00

README.md

PSANet: Point-wise Spatial Attention Network for Scene Parsing

Introduction

@inproceedings{zhao2018psanet,
  title={Psanet: Point-wise spatial attention network for scene parsing},
  author={Zhao, Hengshuang and Zhang, Yi and Liu, Shu and Shi, Jianping and Change Loy, Chen and Lin, Dahua and Jia, Jiaya},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  pages={267--283},
  year={2018}
}

Results and models

Cityscapes

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
PSANet R-50-D8 512x1024 40000 7 3.17 77.63 79.04 config model | log
PSANet R-101-D8 512x1024 40000 10.5 2.20 79.14 80.19 config model | log
PSANet R-50-D8 769x769 40000 7.9 1.40 77.99 79.64 config model | log
PSANet R-101-D8 769x769 40000 11.9 0.98 78.43 80.26 config model | log
PSANet R-50-D8 512x1024 80000 - - 77.24 78.69 config model | log
PSANet R-101-D8 512x1024 80000 - - 79.31 80.53 config model | log
PSANet R-50-D8 769x769 80000 - - 79.31 80.91 config model | log
PSANet R-101-D8 769x769 80000 - - 79.69 80.89 config model | log

ADE20K

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
PSANet R-50-D8 512x512 80000 9 18.91 41.14 41.91 config model | log
PSANet R-101-D8 512x512 80000 12.5 13.13 43.80 44.75 config model | log
PSANet R-50-D8 512x512 160000 - - 41.67 42.95 config model | log
PSANet R-101-D8 512x512 160000 - - 43.74 45.38 config model | log

Pascal VOC 2012 + Aug

Method Backbone Crop Size Lr schd Mem (GB) Inf time (fps) mIoU mIoU(ms+flip) config download
PSANet R-50-D8 512x512 20000 6.9 18.24 76.39 77.34 config model | log
PSANet R-101-D8 512x512 20000 10.4 12.63 77.91 79.30 config model | log
PSANet R-50-D8 512x512 40000 - - 76.30 77.35 config model | log
PSANet R-101-D8 512x512 40000 - - 77.73 79.05 config model | log