PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
.readme static -> .readme Jan 22, 2018
examples/voc Fix for flake8 Oct 25, 2018
tests/models_tests Update tests/models_tests/test_fcn32s.py Nov 17, 2017
torchfcn Changes for latest pytorch Oct 24, 2018
.gitignore Add dirs Feb 25, 2017
.travis.yml Use hacking instead of flake8 Oct 30, 2018
MANIFEST.in Fix MANIFESt.in Apr 12, 2017
README.md Add notice for pretrained model to README.md Aug 28, 2018
requirements.txt Update torch requirement Jun 8, 2018
setup.cfg Add FCN16s Aug 8, 2017
setup.py Update setup.py Jun 8, 2018

README.md

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.