/pytorch-deeplab-xception

DeepLab v3+ model in PyTorch

Primary LanguagePythonMIT LicenseMIT

pytorch-deeplab-xception

Introduction

This is a Pytorch implementation of DeepLab-Xception, It's based on a Modified Aligned Xception backbone. Currently, we train DeepLab V3+ from scratch, using Pascal Voc 2012 dataset.

TODO

  • Basic deeplab v3+ model
  • Training deeplab v3+ on SBD dataset
  • Add pretrained model and results evaluation

Requirements

Python3.x (Tested with 3.5)
PyTorch (Tested with 0.4.0)
tensorboardX
opencv-python

Installation

This code is tested in Ubuntu 16.04. To use this code, please do:

  1. Clone the repo:

    git clone https://github.com/jfzhang95/pytorch-deeplab-xception.git
cd pytorch-deeplab-xception

  2. To train DeepLabV3+, please do:

    python train.py