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 |