Pixel-wise segmentation on the VOC2012 dataset using pytorch.
See dataset examples here.
Download image archive and extract and do:
mkdir data
mv VOCdevkit/VOC2012/JPEGImages data/images
mv VOCdevkit/VOC2012/SegmentationClass data/classes
rm -rf VOCdevkit
We recommend using pyenv:
pyenv virtualenv 3.6.0 piwise
pyenv activate piwise
then install requirements with pip install -r requirements.txt.
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