怎么用C++提取任意一张图片的特征（从内存读取数据）

2024-02-24 15:50:05 155

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

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

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

layer{
name:"data"
type:"MemoryData"
top:"data"
top:"label"
memory_data_param{
batch_size:1
channels:3
height:100
width:100
}
}

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

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

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

#include
#include
#include
#include
#include
#include"boost/algorithm/string.hpp"
#include"google/protobuf/text_format.h"
#include"caffe/blob.hpp"
#include"caffe/common.hpp"
#include"caffe/net.hpp"
#include"caffe/proto/caffe.pb.h"
#include"caffe/util/io.hpp"
#include"caffe/layers/memory_data_layer.hpp"
#defineNetTyfloat
usingnamespacecaffe;
usingstd::cout;
usingstd::endl;
usingstd::string;
/*加载模型函数*/
template
caffe::Net*loadNet(std::stringparam_file,std::stringpretrained_param_file,caffe::Phasephase)
{
caffe::Net*net(newcaffe::Net(param_file,phase));
net->CopyTrainedLayersFrom(pretrained_param_file);
returnnet;
}
intmain()
{
cv::Matsrc;
src=cv::imread("face_example/test.jpg");//读取测试图片
cv::resize(src,src,cv::Size(100,100));//这里要将图片resize到prototxt里面的输入层指定的大小
caffe::Net*_net=loadNet("face_example/face_deploy.prototxt","face_example/face.caffemodel",caffe::TEST);//加载网络定义文件和参数模型
caffe::MemoryDataLayer*m_layer=(caffe::MemoryDataLayer*)_net->layers()[0].get();//定义个内存数据层指针
std::vectordv={src};//AddMatVector(constvector&mat_vector,constvector&labels)
std::vectorlabel={0};//-------------------------------------------------------------------------
m_layer->AddMatVector(dv,label);//把图片和标签，添加到MemoryData层
std::vector*>input_vec;//无意义，为了函数参数需要
_net->Forward(input_vec);//执行一次前向计算
boost::shared_ptr>layerData=_net->blob_by_name("res5_6");//获得指定层的输出
constNetTy*pstart=layerData->cpu_data();//res5_6->cpu_data()返回的是多维数据（数组）
/*-----输出特征-----*/
for(inti=0;i<30000;i++)
{
std::cout<<*pstart<
以上所述是小编给大家介绍的怎么用C++提取任意一张图片的特征（从内存读取数据），希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对毛票票网站的支持！

怎么用C++提取任意一张图片的特征（从内存读取数据）

热门推荐

随机推荐