youdonny/torchcv-seg

Semantic Segmentation ported from PyTorchCV

PyTorchCV-SemSeg: Open Source for Semantic Segmentation.

@misc{CV2018,
  author =       {Donny You (youansheng@gmail.com)},
  howpublished = {\url{https://github.com/donnyyou/PyTorchCV-SemSeg}},
  year =         {2018}
}

This repository provides source code for some deep learning based semantic segmentation. We'll do our best to keep this repository up to date. If you do find a problem about this repository, please raise it as an issue. We will fix it immediately.

Implemented Papers

• Semantic Segmentation
  • PSPNet: Pyramid Scene Parsing Network
  • DeepLabV3: Rethinking Atrous Convolution for Semantic Image Segmentation
  • DenseASPP: DenseASPP for Semantic Segmentation in Street Scenes

Performances with PyTorchCV-SemSeg

• CityScapes (Single Scale Whole Image Test): Base LR 0.01, Crop Size 769

Checkpoints	Backbone	Train	Test	mIOU	BS	Iters	Scripts
PSPNet	3x3-Res101	train	val	-	8	4W	PSPNet
DeepLabV3	3x3-Res101	train	val	-	8	4W	DeepLabV3

• ADE20K (Single Scale Whole Image Test): Base LR 0.02, Crop Size 520

Checkpoints	Backbone	Train	Test	mIOU	PixelACC	BatchSize	Iters	Scripts
PSPNet	3x3-Res50	train	val	-	-	16	15W	PSPNet
DeepLabv3	3x3-Res50	train	val	-	-	16	15W	DeepLabV3
PSPNet	3x3-Res101	train	val	-	-	16	15W	PSPNet
DeepLabv3	3x3-Res101	train	val	-	-	16	15W	DeepLabV3