Home

Awesome

pytorch-fcn

PyPI Version Python Versions GitHub Actions

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.

ModelImplementationepochiterationMean IUPretrained Model
FCN32sOriginal--63.63Download
FCN32sOurs119600062.84
FCN16sOriginal--65.01Download
FCN16sOurs119600064.91
FCN8sOriginal--65.51Download
FCN8sOurs76000065.49
FCN8sAtOnceOriginal--65.40
FCN8sAtOnceOurs119600064.74
<img src=".readme/fcn8s_iter28000.jpg" width="50%" /> Visualization of validation result of FCN8s.

Cite This Project

If you use this project in your research or wish to refer to the baseline results published in the README, please use the following BibTeX entry.

@misc{pytorch-fcn2017,
  author =       {Ketaro Wada},
  title =        {{pytorch-fcn: PyTorch Implementation of Fully Convolutional Networks}},
  howpublished = {\url{https://github.com/wkentaro/pytorch-fcn}},
  year =         {2017}
}