Caffe Windows系列(2): 使用C++ API进行分类

由于我的目标就是用Caffe来做图像的分类,而且是要用C++来做图像的分类。那么,目前最重要的事就是要验证一下能不能用Caffe做这件事,以及Caffe是怎么做这件事的。看到官网上提供了一个例子:Classifying ImageNet: using the C++ API。那我们就从这里开始吧。](http://caffe.berkeleyvision.org/gathered/examples/cpp_classification.html)。那我们就从这里开始吧。)

Caffe, at its core, is written in C++. It is possible to use the C++ API of Caffe to implement an image classification application similar to the Python code presented in one of the Notebook examples. To look at a more general-purpose example of the Caffe C++ API, you should study the source code of the command line tool caffe in tools/caffe.cpp.

A simple C++ code is proposed in examples/cpp_classification/classification.cpp.

也就是,官网说Caffe是完全可以用C++接口进行调用的。要想知道怎么调用,可以研究一下tools/caffe.cpp和examples/cpp_classification/classification.cpp这两个文件。按我目前的理解,tools/caffe.cpp是用来训练的,而classification.cpp则是教我们如何用已经训练好的模型进行分类。其实,对于caffe可以训练这件事是毋庸置疑的,只不过还没有深入了解。但现在最重要的,是要看看当模型训练完成之后,如何调用C++ API函数对图像进行识别。所以,本文章仅对examples/cpp_classification/classification.cpp这个文件感兴趣。网上已经有一个大神写了一篇关于这个议题的文章了,因此,我就只是在Windows系统上来重新实验一下吧。

目标

在Windows 7上实验对小猫图片(examples\images\cat.jpg)进行分类。

准备

我们需要准备三个文件:

  • caffemodel文件:可以直接在此地址下载,下载后放到models\bvlc_reference_caffenet文件夹下。
  • 均值文件: notepad++打开data/ilsvc12/get_ilsvrc_aux.sh,“编辑”->“文档格式转换”->“转换为UNIX格式”。在caffe目录下执行
data/ilsvc12/get_ilsvrc_aux.sh

执行并下载后,均值文件放在 data/ilsvrc12/ 文件夹里。

  • synset_words.txt文件
    在调用脚本文件下载均值的时候,这个文件也一并下载好了。里面放的是1000个类的名称。

C++实现

用Visual Studio打开Caffe.sln。生成examples\classification。如果生成的是Release版本,则生成的exe程序会在caffe\build\examples\cpp_classification\Release中。在控制台运行classification.exe,出现了Usage解释:

Usage: classification.exe deploy.prototxt network.caffemodel mean.binaryproto la
bels.txt img.jpg

也就是说一共需要五个参数:

  1. deploy.prototxt
  2. network.caffemodel
  3. mean.binaryproto
  4. labels.txt
  5. img.jpg
> cd caffe
> build\examples\cpp_classification\Release\classification.exe ^
  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

输出结果

---------- Prediction for examples\images\cat.jpg ----------
0.3134 - "n02123045 tabby, tabby cat"
0.2380 - "n02123159 tiger cat"
0.1235 - "n02124075 Egyptian cat"
0.1003 - "n02119022 red fox, Vulpes vulpes"
0.0715 - "n02127052 lynx, catamount"

即有0.3134的概率为tabby cat, 有0.2380的概率为tiger cat ......

OK,到此为止,说明用C++写程序来预测某一张图像属于哪个类是可用的。但如果真的自己要写的话,还是得仔细看一下classification.cpp。它一共有265行,我们就先来看一下它的main函数吧。

int main(int argc, char** argv) {
  if (argc != 6) {
    std::cerr << "Usage: " << argv[0]
              << " deploy.prototxt network.caffemodel"
              << " mean.binaryproto labels.txt img.jpg" << std::endl;
    return 1;
  }
 
  ::google::InitGoogleLogging(argv[0]);
 
  string model_file   = argv[1];
  string trained_file = argv[2];
  string mean_file    = argv[3];
  string label_file   = argv[4];
  Classifier classifier(model_file, trained_file, mean_file, label_file);
 
  string file = argv[5];
 
  std::cout << "---------- Prediction for "
            << file << " ----------" << std::endl;
 
  cv::Mat img = cv::imread(file, -1);
  CHECK(!img.empty()) << "Unable to decode image " << file;
  std::vector<Prediction> predictions = classifier.Classify(img);
 
  /* Print the top N predictions. */
  for (size_t i = 0; i < predictions.size(); ++i) {
    Prediction p = predictions[i];
    std::cout << std::fixed << std::setprecision(4) << p.second << " - \""
              << p.first << "\"" << std::endl;
  }
}

line# 2-7: 显示用法信息
line# 9: 初始化glog……很惭愧,还没有使用过glog
line# 15: 初始化classifier对象。Classifier类是关键,也是在这个文件中定义的。
line# 22: 用OpenCV库读取图像
line# 24: 调用classifier的方法Classify,对刚刚读取的图像进行识别,并将结果存储在Prediction数组中,其中,Prediction的定义为

typedef std::pair<string, float> Prediction;

是<label,score>键值对。
line# 27-31: 输出结果。

cpp除了main以外的其它地方就干了一件事:定义Classifier。等到真正要用的时候,我们可以把Classifier类直接拿来用。因此,暂时就不展开分析了。

总结

如果我们有了caffemodel文件均值文件synset_words.txt文件,则借助** Classifier类**,就可以实现简单地对OpenCV Mat图像预测,给出前n个预测结果。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容