Hugging Face's logo

mccaly
/
test2

Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
Model card Files Community
2
test2 / configs /pspnet /README.md
mccaly's picture
mccaly
Upload 660 files
b13b124
preview code
|
raw
history blame
15.8 kB

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