/pytorch-fcn

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

Primary LanguagePython

pytorch-fcn

PyPI Version Python Versions Build Status

PyTorch implementation of Fully Convolutional Networks.

Requirements

Installation

git clone https://github.com/wkentaro/pytorch-fcn.git
cd pytorch-fcn
pip install .

# or

pip install torchfcn

Training

See VOC example.

Accuracy

At 10fdec9.

Model Implementation epoch iteration Mean IU Pretrained Model
FCN32s Original - - 63.63 Download
FCN32s Ours 11 96000 62.84
FCN16s Original - - 65.01 Download
FCN16s Ours 11 96000 64.91
FCN8s Original - - 65.51 Download
FCN8s Ours 7 60000 65.49
FCN8sAtOnce Original - - 65.40 Download
FCN8sAtOnce Ours 11 96000 64.74

Visualization of validation result of FCN8s.