1. 程式人生 > >用caffe訓練LeNet網路的步驟

用caffe訓練LeNet網路的步驟

以mnist手寫體資料為例

1.準備資料

cd $CAFFE_ROOT
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

2.定義Net
$CAFFE_ROOT/examples/mnist/lenet_train_test.prototxt

3.配置Solver
$CAFFE_ROOT/examples/mnist/lenet_solver.prototxt

4.訓練和測試
cd $CAFFE_ROOT
./examples/mnist/train_lenet.sh