Ubuntu14.04+caffe实现cifar10例程及单张图片测试

一.环境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    (此处可以是自己的图片)

你可能感兴趣的:(Ubuntu14.04+caffe实现cifar10例程及单张图片测试)