最近準(zhǔn)備從 Caffe 入手,學(xué)習(xí)下深度學(xué)習(xí)的基本知識(shí)盈滴,于是便在筆記本電腦上的 Deepin 系統(tǒng)中進(jìn)行了一些實(shí)踐巢钓,其中參考了網(wǎng)上不少資料症汹,特此記錄下來(lái)背镇。
* 系統(tǒng)配置:
顯卡:Intel 集成顯卡
操作系統(tǒng):Deepin 15.4.1 x64
* 安裝步驟:
一瞒斩、安裝 OpenCV 3.3.0
(1)安裝 OpenCV 所需要的庫(kù)胸囱,在終端依次執(zhí)行以下命令:
sudo apt-get update
sudo apt-get install build-essential
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev
(2)官網(wǎng)下載 OpenCV 3.3.0 的源代碼(下載地址:https://opencv.org/releases.html)并解壓到 opencv3.3.0 目錄烹笔,然后進(jìn)入到 opencv3.3.0 的目錄箕宙,執(zhí)行:
mkdir build
cd build
(3)配置 OpenCV 工程文件
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
如果執(zhí)行命令時(shí)出現(xiàn)下載 ippicv 失敗铺纽,可以參考文章解決:http://blog.csdn.net/neilooo/article/details/78425559。
(4)編譯安裝
sudo make install
這個(gè)過(guò)程會(huì)有點(diǎn)久锅很,完成之后 OpenCV3.3.0 就安裝成功了凤跑。
二仔引、安裝 Caffe
(1)安裝 Caffe 依賴庫(kù)
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
(2)安裝 BLAS
sudo apt-get install libatlas-base-dev
(3)安裝其余依賴庫(kù)
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
(4)下載 Caffe 代碼到主目錄下
sudo apt-get install cmake git
git clone https://github.com/BVLC/caffe.git
(5)進(jìn)入 Caffe 目錄咖耘,配置 Makefile.config
cd caffe/
cp Makefile.config.example Makefile.config
gedit Makefile.config
以下內(nèi)容根據(jù)個(gè)人實(shí)際情況修改:
僅使用 CPU
# CPU_ONLY := 1
改為
CPU_ONLY := 1
使用 OpenCV3
# OPENCV_VERSION := 3
改為
OPENCV_VERSION := 3
使用 Python Layer
# WITH_PYTHON_LAYER := 1
改為
WITH_PYTHON_LAYER := 1
增加依賴的頭文件和庫(kù)路徑:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改為
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial /usr/liclude /usr/include/leveldb /usr/local/include/gflags /usr/include/google/protobuf
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr /usr/lib/x86_64-linux-gnu/hdf5/serial
保存并關(guān)閉文本版保。
(5)修改 Makefile 文件
gedit Makefile
將里面的
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
修改為
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
保存并關(guān)閉文本彻犁。
(6)編譯 Caffe
make all
make test
make runtest
命令全部正確執(zhí)行后汞幢,Caffe 即安裝完畢急鳄。
三堰酿、安裝 Pycaffe(可選)
(1)安裝 pip 并換鏡像源
sudo apt-get install python-pip
sudo gedit /etc/pip.conf
在打開(kāi)的文本文件中輸入以下內(nèi)容并保存触创。
[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
(2)安裝依賴包
pip install -U scikit-image
pip install protobuf
(3)將 Caffe 的 Python 路徑加入環(huán)境變量:
gedit ~/.bashrc
在文件末尾添加一行:
export PYTHONPATH="/home/YOUR-NAME/caffe/python:$PYTHONPATH"
其中,YOUR-NAME 用真實(shí)的用戶名替代抖韩,保存并關(guān)閉文本疫铜,然后執(zhí)行:
sudo ldconfig
make pycaffe
接著關(guān)閉當(dāng)前終端并重新開(kāi)啟一個(gè)以使環(huán)境變量生效。
(4)在新的終端輸入 python 以進(jìn)入 python 命令行模式顽馋,然后輸入 import caffe寸谜,如果沒(méi)有錯(cuò)誤熊痴,則 Pycaffe 安裝成功。
* 用例測(cè)試:
下面以入門(mén)級(jí)的 MNIST 手寫(xiě)數(shù)字?jǐn)?shù)據(jù)集為例做一個(gè)測(cè)試果善。
(1)首先進(jìn)入 Caffe 目錄岭埠,打開(kāi)文件 examples/mnist/lenet_solver.prototxt
cd caffe/
gedit examples/mnist/lenet_solver.prototxt
將其中的
solver_mode: GPU
改為
solver_mode: CPU
保存并關(guān)閉文本惜论。
(2)依次輸入以下指令:
# 下載 MNIST 數(shù)據(jù)集
./data/mnist/get_mnist.sh
# 數(shù)據(jù)格式轉(zhuǎn)換
./examples/mnist/create_mnist.sh
# 訓(xùn)練 LeNet-5 超參數(shù)
./examples/mnist/train_lenet.sh
# 對(duì)測(cè)試集進(jìn)行預(yù)測(cè)測(cè)試
./build/tools/caffe.bin test -model examples/mnist/lenet_train_test.prototxt \
-weights examples/mnist/lenet_iter_10000.caffemodel -iterations 100
用例測(cè)試完成馆类,開(kāi)始 Caffe 之旅吧!