Caffe之旅（一）：训练mnist

根据教程 https://blog.csdn.net/city_to_sky/article/details/78977353安装完caffe后应该会在caffe_root\examples下有一个minist文件夹

进入mnist文件夹后会有三个.ps1文件，也就是powershell执行文件

其中create_mnist.ps1内容如下

$CaffeRoot = (Resolve-Path (Join-Path $PSScriptRoot ..\..)) 
$EXAMPLE = "$CaffeRoot\examples\mnist"
$DATA = "$CaffeRoot\data\mnist"
if("$BuildDir" -eq "") {
  $BuildDir = "$CaffeRoot\build"
}
$BUILD = "$BuildDir\examples\mnist\Release"

$BACKEND = "lmdb"

echo "Creating $BACKEND..."

if(Test-Path $EXAMPLE\mnist_train_$BACKEND) {
  rm -Recurse -Force $EXAMPLE\mnist_train_$BACKEND
}
if(Test-Path $EXAMPLE\mnist_train_$BACKEND) {
  rm -Recurse -Force $EXAMPLE\mnist_test_$BACKEND
}

. $BUILD\convert_mnist_data.exe $DATA\train-images.idx3-ubyte `
  $DATA\train-labels.idx1-ubyte $EXAMPLE\mnist_train_$BACKEND --backend=$BACKEND
. $BUILD\convert_mnist_data.exe $DATA\t10k-images.idx3-ubyte `
  $DATA\t10k-labels.idx1-ubyte $EXAMPLE\mnist_test_$BACKEND --backend=$BACKEND

echo "Done."

可以看出，这个脚本文件是将CaffeRoot/data/mnist里的数据文件转换为caffe可以识别的mdb格式存放在mnist_train_lmdb文件夹和mnist_test_lmdb文件夹下。使用powershell运行后可以看到在脚本所在的文件夹里出现了刚才提到的两个文件夹：

然后再看train_lenet.ps1内容

param(
  [string]$BuildDir
)

$CaffeRoot = (Resolve-Path (Join-Path $PSScriptRoot ..\..))
if("$BuildDir" -eq "") {
  $BuildDir = "$CaffeRoot\build"
}

. $BuildDir\install\bin\caffe.exe train --solver=lenet_solver.prototxt $args

可以看出，该文件就是调用caffe来对mnist数据进行训练，其中的lenet_solver.prototxt参数如果感兴趣的童鞋可以参见我的下一篇博客

训练完毕后再使用test_lenet.ps1即可查看最终训练的精度怎样