Ubuntu 16.4下安裝caffe及手寫識(shí)別

Ubuntu 16.4下安裝caffe

1.安裝caffe所需要的依賴包

sudo apt-get install git
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

2.下載Caffe源碼:

git clone https://github.com/bvlc/caffe.git
cd caffe/
mv Makefile.config.example Makefile.config

3.修改Makefile和Makefile.config兩個(gè)文件

修改:Makefile.config:(95行左右)
INCLUDE_DIRS := /usr/include $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/
LIBRARY_DIRS :=/usr/lib $(PYTHON_LIB) /usr/local/lib /usr/lib

打開CPU_ONLY := 1(入門建議先只用cpu訓(xùn)練,進(jìn)階再用gpu)
修改:Makefile:(181行左右)
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

4.在caffe根目錄:

make all -j4

5.運(yùn)行手寫識(shí)別過程:在caffe目錄下

cd data/mnist
./get_mnist.sh(若沒有權(quán)限,chmod +x ./get_mnist.sh 賦予權(quán)限) 下載數(shù)據(jù)

6.轉(zhuǎn)換格式

./examples/mnist/create_mnist.sh
7.訓(xùn)練超參數(shù)
在/examples/mnist/lenet_solver.prototxt將mode改為cpu
再運(yùn)行
./build/tools/caffe train --solver=examples/mnist/lenet_solver.prototxt $@
可以看到最后的結(jié)果達(dá)到0.9917

I0705 17:10:00.829900 17611 data_layer.cpp:73] Restarting data prefetching from start.
I0705 17:10:00.991636 17608 solver.cpp:398]     Test net output #0: accuracy = 0.9917
I0705 17:10:00.991662 17608 solver.cpp:398]     Test net output #1: loss = 0.0268429 (* 1 = 0.0268429 loss)
I0705 17:10:00.991664 17608 solver.cpp:316] Optimization Done.
I0705 17:10:00.991667 17608 caffe.cpp:259] Optimization Done.

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容