资源分类
》 资源 》算法》pytorch-semantic-segmentation

pytorch-semantic-segmentation

2019-12-19 | |  175 |   0 |   0

PyTorch for Semantic Segmentation

This repository contains some models for semantic segmentation and the pipeline of training and testing models, implemented in PyTorch

Models

  1. Vanilla FCN: FCN32, FCN16, FCN8, in the versions of VGG, ResNet and DenseNet respectively (Fully convolutional networks for semantic segmentation)

  2. U-Net (U-net: Convolutional networks for biomedical image segmentation)

  3. SegNet (Segnet: A deep convolutional encoder-decoder architecture for image segmentation)

  4. PSPNet (Pyramid scene parsing network)

  5. GCN (Large Kernel Matters)

  6. DUC, HDC (understanding convolution for semantic segmentation)

Requirement

  1. PyTorch 0.2.0

  2. TensorBoard for PyTorch. Here  to install

  3. Some other libraries (find what you miss when running the code :-P)

Preparation

  1. Go to models directory and set the path of pretrained models in config.py

  2. Go to datasets directory and do following the README

TODO

  1. DeepLab v3

  2. RefineNet

  3. More dataset (e.g. ADE)


上一篇：image-augmentation-in-action

下一篇：awesome-semantic-segmentation-pytorch

用户评价
全部评价

热门资源

  • allennlp-server

    allennlp-server Serve allennlp services as sep...

  • ubuntu-allennlp

    ubuntu-allennlp AllenAI AllenNLP image based o...

  • allennlp_extras

    allennlp_extras Some utilities build on top of...

  • allennlp-dureader

    An Apache 2.0 NLP research library, built on Py...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...