运行caffe自带的mnist实例教程
生活随笔
收集整理的這篇文章主要介紹了
运行caffe自带的mnist实例教程
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
運行caffe自帶的mnist實例教程
1、先進入caffe文件目錄,(指令:cd ./caffe),再用data/mnist下的get_mnist.sh下載MNIST數據集,代碼如下:
2、轉換格式,代碼如下:
./examples/mnist/create_mnist.sh3、訓練超參數,有兩種方式:
-
(1)通過命令行執行訓練,代碼如下:(這是書上的例子,可以參考網址上的例子,在后面。不同之處:網址上的例子在此處新建一個文件夾保存訓練的模型)
-
2)將以上訓練的命令行代碼寫成訓練腳本并命名為train_lenet.sh,放到mnist目錄下。運行train_lenet.sh腳本進行訓練。
5、用訓練好的模型對數據進行預測。
./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_5000.caffemodel -iterations=100總結
以上是生活随笔為你收集整理的运行caffe自带的mnist实例教程的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Ubuntu 16.04 安装 caff
- 下一篇: caffe的prototxt文件