pytorch-fcn

PyTorch implementation of Fully Convolutional Networks.

Requirements

• pytorch >= 0.2.0
• torchvision >= 0.1.8
• fcn >= 6.1.5
• Pillow
• scipy
• tqdm

Installation

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

conda install pytorch cuda80 torchvision -c soumith
pip install .

Training

See VOC example.

Accuracy

At 10fdec9.

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

Visualization of validation result of FCN8s.