一.环境Ubuntu14.04+caffe(无GPU)
二.训练模型
下载数据:cd caffe(caffe根目录)
./data/cifar10/get_cifar10.sh
./examples/cifar10/create_cifar10.sh
训练: (注:如果电脑没有GPU,需要更改两个地方,即cifar10_quick_solver.prototxt和cifar10_quick_solver_lr1.prototxt。将其中最后一行的GPU改为CPU。(文件在caffe/examples/cifar10下))
cd caffe(同样在根目录下)
./examples/cifar10/train_quick.sh
训练完毕之后,cifar10文件夹下会生成四个文件。
三.利用classification.bin识别单张图片 注:每一行的反斜杠之前都有一个空格
回到根目录下(caffe目录下 cd caffe)执行:
$ ./build/examples/cpp_classification/classification.bin \
>examples/cifar10/cifar10_quick.prototxt \ (模型描述文件)
>examples/cifar10/cifar10_quick_iter_5000.caffemodel.h5 \ (训练好的caffemodel模型权值文件)
>examples/cifar10/mean.binaryproto \ (图像均值文件)
>data/cifar/batches.meta.txt \ (图像类别标签信息) 注:此标签可能是11类,因为最后一行有一个空格,需要删掉,否则和模型描述文件不符。
>examples/images/fish-bike.jpg (此处可以是自己的图片)