1. 程式人生 > >ubuntu下caffe實戰---mnist資料集測試

ubuntu下caffe實戰---mnist資料集測試

測試用到手寫體識別模型LeNet,mnist資料集

其中LeNet是一種CNN模型,由一個卷積層、後面跟一個下采樣層、再跟另外一個卷積層和另一個下采樣層,再之後是兩個全連線層組成。

mnist資料集:包括60000個訓練集和10000個驗證集

訓練過程:

1.下載mnist資料集

cd ~/caffe

./data/mnist/get_mnist.sh

如果出現許可權不夠的情況,就加個chmod +X賦予許可權

2.轉換格式

下載的資料集為二進位制格式,必須轉換成caffe能識別的lmdb格式

./examples/mnist/create_mnist.sh

瀏覽/example/mnist下出現mnist_train_lmdb和mnist_test_lmdb就對了

3.訓練資料集

./example/mnist/train_lenet.sh

相信大家在網上看到的都是這麼幾步,我也沒有copy,純手打,主要因為錯了好幾次,也敲了好幾次,已經印在腦海中了。

主要說一些常見的問題和解決辦法:

a.找不到caffe,解決辦法

sudo gedit ~/.bashrc 
PYTHONPATH=/home/sunny/caffe-master/python:$PYTHONPATH 
sudo ldconfig 

b.執行./example/mnist/train_lenet.sh時提示/examples/build/tools/caffe:No such directory or file,解決辦法:

開啟/example/mnist/train_lenet.sh,把絕對路徑註釋掉改為相對路徑

c.好不容易把第二個問題解決了,能訓練了,列印訓練引數網路結束後卻提示:已放棄,核心已轉存

這時候只需要在./example/mnist/train_lenet.sh前加個sudo就解決問題了

最後訓練完的結果是這樣的:

分類準確率99.11%,達到訓練效果。

很多人到第三步就完了

4.用訓練好的模型對資料進行預測

 ./build/tools/caffe.bin test \

 -model examples/mnist/lenet_train_test.prototxt \

-weights examples/mnist/lenet_iter_10000.caffemodel \

 iterations 100

轉載請註明出處