Implement some models of semantic segmentation in PyTorch, easy to run.
- This repo includes some networks for Semantic Segmentation implemented in pytorch 1.0.0 and python3. See each directory for more information.
- I only provide architecture of network here. Dataset and train/test files aren't available here, for I think it can be added according to individual needs.
- The code file containing the network structure can be run directly with the set simulation data.
- ResNet101 used in this repo is the one which PSPNet used. The difference between this resnet and the original resnet is that the first 7*7 conv layer in the old version is replaced by three small-kernel convs. Pretrained model can be downloaded from here.