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
轉載請註明出處