PSPNet in Chainer
This is an unofficial implementation of Pyramid Scene Parsing Network (PSPNet) in Chainer.
pip install chainer --pre
pip install cupy --pre
pip install git+git://github.com/chainer/chainermn
pip install git+git://github.com/chainer/chainercv
pip install tqdm
$ python demo.py -g 0 -m cityscapes -f aachen_000000_000019_leftImg8bit.png
$ python demo.py -g 0 -m voc2012 -f 2008_000005.jpg
$ python demo.py -g 0 -m ade20k -f ADE_val_00000001.jpg
If you get RuntimeError: Invalid DISPLAY variable
, how about specifying the matplotlib's backend by an environment variable?
$ MPLBACKEND=Agg python demo.py -g 0 -m cityscapes -f aachen_000000_000019_leftImg8bit.png
Caffe is NOT needed to convert .caffemodel
to Chainer model. Use caffe_pb2.py
.
Please download the weights below from the author's repository:
and then put them into weights
directory.
$ python convert.py