Github链接:https://github.com/kazuto1011/pspnet-pytorch
Requirements
pytorch
click
addict
pydensecrf
protobuf
pyyaml
下载原作者的caffe.proto到libs文件夹(该项目里已经下载好),再下载预训练权重(需科学上网)
地址如下:
ADE20K: https://drive.google.com/open?id=0BzaU285cX7TCN1R3QnUwQ0hoMTA
VOC2012: https://drive.google.com/open?id=0BzaU285cX7TCNVhETE5vVUdMYk0
cityscapes: https://drive.google.com/open?id=0BzaU285cX7TCT1M3TmNfNjlUeEU
cd到该项目所在文件夹,按照作者所给步骤运行即可。