2016年Semantic Segmentation方向比较出色的一篇文章,DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs
https://arxiv.org/abs/1606.00915
作者提供了开源的代码,shell版本:https://bitbucket.org/aquariusjay/deeplab-public-ver2
和python版本:https://github.com/TheLegendAli/DeepLab-Context2
以及相关Model下载:
http://liangchiehchen.com/projects/DeepLab_Models.html
作者所使用的Caffe版本比较陈旧,导致会出现很多和最新环境不兼容的情况。我使用的环境是Ubuntu 16.04, CUDA 8.0, cuDNN 5.1, Python 2.7.12,以下就以这个配置写个简单流程说明。
1.安装
1.1 安装Matio
下载matio(https://sourceforge.net/projects/matio/files/matio/1.5.2/)
$ tar zxf matio-1.5.2.tar.gz
$ cd matio-1.5.2
$ ./configure
$ make
$ make check
$ make install
$ sudo ldconfig
1.2 安装wget
执行脚本时如果提示 ImportError: No module named wget,说明还需要安装wget
sudo pip install wget
2.编译
和BVLC版本一样,对DeepLab版本的caffe进行编译