Colorful Image Colorization 的环境配置

原文链接：https://github.com/richzhang/colorization
步骤基本是按照Installation里的说明

1.安装依赖库

1.1Python相关库

Python libraries (numpy, pyplot, skimage, scipy)。相关命令：

sudo apt-get install python
sudo apt-get install python-numpy
sudo apt-get install python-matplotlib
sudo apt-get install python-scipy

1.2python的ipython notebook

• Ubuntu ipython notebook的安装
• ipython notebook简介

2.安装caffe

安装步骤主要参考Caffe在Ubuntu 14.04 64bit 下的安装和caffe官方安装文档
说明：要根据自己机器的情况在Makefile.config里设置

• 依赖项
  sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
• 安装ATLASsudo apt-get install libatlas-base-dev
• 编译。由于我只用CPU的集成显卡所以注意官方说明里的For CPU-only Caffe, uncomment CPU_ONLY := 1 in Makefile.config. 然后按下列命令编译

cp Makefile.config.example Makefile.config
# Adjust Makefile.config (按照机器情况修改)
make all
make test
make runtest

• 测试。终端里命令如下。如果没有报错即为安装成功

python
import caffe

3.具体配置

基本按照项目网站的Installation设置

• follow项目
  git clone -b master --single-branch https://github.com/richzhang/colorization.git
• cd 到demo目录里cd ~/colorization/demo
• 运行ipython notebook ipython notebook。会自动提取demo目录里的colorization_demo_v0.ipynb并在浏览器里运行。鼠标点击相应的代码点击运行按钮即为运行相应代码块。