caffe初探之-使用C++ API 对 ImageNet 分类

来源:互联网 发布:京东聊天软件 编辑:程序博客网 时间:2024/06/05 08:13
本文示例来自caffe官网 地址 http://caffe.berkeleyvision.org/gathered/examples/cpp_classification.html

1.准备

一个简单的C++ 代码例子 examples/cpp_classification/classification.cpp 

2.编译

这个C++示例在编译caffe时自动生成的。classification.bin文件已经在安装编译caffe时建立,所在位置:

caffe/build/examples/cpp_classification/classification.bin

3.使用

使用已经训练好的CaffeNet 模型来进行分类,这个模型在从“Model Zoo”上下载

 ./scripts/download_model_binary.py models/bvlc_reference_caffenet
下载 mageNet 的标签文件
./data/ilsvrc12/get_ilsvrc_aux.sh
注:在下载bvlc_reference_caffenet.caffemodel 时总是失败,就在这下了http://dl.caffe.berkeleyvision.org/bvlc_reference_caffenet.caffemodel
对下面这只猫进行分类

./build/examples/cpp_classification/classification.bin models/bvlc_reference_caffenet/deploy.prototxt models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel data/ilsvrc12/imagenet_mean.binaryproto data/ilsvrc12/synset_words.txt examples/images/cat.jpg

自己下了一张图片试了一下

输出结果


0 0
原创粉丝点击