Skip to main content

Fully Convolutional Networks

Project description

https://badge.fury.io/gh/wkentaro%2Ffcn.svg https://travis-ci.org/wkentaro/fcn.svg?branch=master

This is Chainer implementation of fcn.berkeleyvision.org.

Features

  • Provide FCN8s model for Chainer. [v1.0.0]

  • Copy caffemodel to chainermodel. [v1.0.0]

  • Forwarding with Chainer for pascal dataset. [v1.0.0]

  • Training with Chainer for pascal dataset. [v2.0.0]

  • Training for APC2015 dataset. [v3.0.0]

License

Copyright (C) 2016 Kentaro Wada
Released under the MIT license

Installation

# Ubuntu: install required libraries via apt
sudo apt-get install liblapack-dev   # for numpy
sudo apt-get install libhdf5-dev     # for h5py
# macOS: install required libraries via brew
brew install gfortran  # for numpy
brew install hdf5      # for h5py

pip install fcn

For Inference

Inference is done as below:

# Download sample image
wget https://farm2.staticflickr.com/1522/26471792680_a485afb024_z_d.jpg -O sample.jpg

# forwaring of the networks
fcn_infer.py --img-files sample.jpg --gpu -1  # cpu mode
fcn_infer.py --img-files sample.jpg  # gpu mode
static/fcn8s_26471792680.jpg

Original Image: https://www.flickr.com/photos/faceme/26471792680/

For Developing

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

python setup.py develop

Training

cd examples/pascal
./download_dataset.py

./train_fcn32s.py

Currently we support only training FCN32s. The learning curve looks like below:

static/fcn32s_learning_curve.png

Inference with FCN32s + 60000 iterations outputs below result:

static/fcn32s_2007_000129.jpg

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

fcn-5.0.1.tar.gz (1.2 MB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page