caffe-pspnet-ubuntu16.04-CUDA8.0-8G

来源:互联网 发布:雨人医药软件 编辑:程序博客网 时间:2024/06/09 20:03

github: https://github.com/hszhao/PSPNet

运行之前,最好先跑通Deeplab v2, 因为该算法是基于deeplab改进的,配置相似。

安装:
安装前提,配置好caffe+deeplab v2

1.复制代码

git clone https://github.com/hszhao/PSPNet.git

2.编译caffe和matcaffe

cd $PSPNET_ROOTcp Makefile.config.example Makefile.configvim Makefile.configmake -j8 && make matcaffe
  1. 评价
    评价的代码在'evaluation'文件夹下
    下载模型放到'evaluation/model'文件夹下

修改eval_all.m中相关路径,

cd evaluationvim eval_all.m

数据跟目录‘data_root’,评估图像list文件放在’evaluation/samplelist’下。
修改GPU信息,多GPU就用parfor,但GPU就用for
Matlab的parfor并行编程,能够非常方便的在多核机器或集群上实现并行计算

    4.