利用C++读取图片数据内存，一键智能识别技术！

关于使用C++接口来提取特征，caffe官方提供了一个extract_features.cpp的例程，但是这个文件的输入是blob数据，即使输入层使用的是ImageData，也需要在deploy.prototxt中指定图片的位置，很不方便。

image

如果想要使用opencv来读取一个图片，然后用caffe训练好的model提取特征，就需要对输入层进行改写。另外官方例程默认的输出是leveldb格式，我们也可以获取float类型的多维特征（数组），这样集成到我们的项目中更灵活。

这个是小编自己整理的一份C/C++资料，加小编C/C++编程学习qq群：825414254

image

01

首先我们需要改写deploy.prototxt的输入层为"MemoryData"：

image

在之前的训练中可能使用的是"ImageData"、"Data"之类的，现在改成MemoryData不影响。

02

我准备提取的层的名字是"res5_6"，就是"InnerProduct"的前一层，当我想提取"InnerProduct"全连接层的输出时，总是报错，提示原始参数和网络参数不匹配（就是训练好的model和现在deploy的网络维度不一样），所以只好提取前一层了，并且要把全连接层屏蔽掉，屏蔽的方法是把prototxt里相应层的名字改掉就好（相对于caffemodel里面的名字）。[以上问题暂时还没解决，留坑]

03

下面是更改之后的 extract_features.cpp的代码：

image

以上所述是小编给大家介绍的怎么用C++提取任意一张图片的特征（从内存读取数据），希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对小编的支持！

image