README.md

Pyramid Scene Parsing Network

Introduction

@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-50-D8	512x1024	80000	-	-	78.55	79.79	model | log
PSPNet	R-101-D8	512x1024	80000	-	-	79.76	81.01	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

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