PiWiSe

Pixel-wise segmentation on the VOC2012 dataset using pytorch.

Setup

See dataset examples here.

Download

Download image archive and extract and do:

mkdir data
mv VOCdevkit/VOC2012/JPEGImages data/images
mv VOCdevkit/VOC2012/SegmentationClass data/classes
rm -rf VOCdevkit

Install

We recommend using pyenv:

pyenv virtualenv 3.6.0 piwise 
pyenv activate piwise

then install requirements with pip install -r requirements.txt.

Usage

If you want to have visualization open an extra tab with:

python -m visdom.server -port 5000

Then start training:

python main.py --model basic --visualize --port 5000

or if you have a large CUDA card:

python main.py --model unet --cuda --visualize --port 5000