digits工具是caffe的非常给力的预处理工具,开源地址为https://github.com/NVIDIA/DIGITS
安装在Ubuntu16.0.4上安装命令如下所示:
源代码安装方式
1.安装依赖包:
sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-pip python-protobuf python-scipy
2.git程序到本地
DIGITS_ROOT=~/digits git clone https://github.com/NVIDIA/DIGITS.git $DIGITS_ROOT
3.安装软件依赖
sudo pip install -r $DIGITS_ROOT/requirements.txt
4.支持插件(可选)
sudo pip install -e $DIGITS_ROOT
5.添加CAFFE_ROOT环境变量(假设caffe的目录在/home/work/caffe)
export CAFFE_ROOT=/home/work/caffe6.启动digits的服务
./digits-devserver
7.在浏览器中查看
http://localhost:5000/